diff --git a/test/builds/build_pybind11_module.sh b/test/builds/build_pybind11_module.sh
index 701910431..f88e4975b 100644
--- a/test/builds/build_pybind11_module.sh
+++ b/test/builds/build_pybind11_module.sh
@@ -28,6 +28,6 @@ build_pybind11_module() {
 	  -shared \
 	  -fPIC \
 	  -o $MODULE_FILE \
-	  -DMNISTCNNCPP_MODULE_NAME=$MODULE_NAME \
+	  -DPYTHON_MODULE_NAME=$MODULE_NAME \
 	  $CPP_FILE
 }
diff --git a/test/ksc/adbench-lstmpy.cpp b/test/ksc/adbench-lstmpy.cpp
index 3ab105710..a38662b5b 100644
--- a/test/ksc/adbench-lstmpy.cpp
+++ b/test/ksc/adbench-lstmpy.cpp
@@ -36,7 +36,7 @@ void declare_vec(py::module &m, std::string typestr) {
 // possible because Python tuples map to std::tuples regardless of
 // their contents.  I'll look into it later.  For now I'll just have a
 // bunch of verbose replication.
-PYBIND11_MODULE(MNISTCNNCPP_MODULE_NAME, m) {
+PYBIND11_MODULE(PYTHON_MODULE_NAME, m) {
   declare_vec<double>(m, std::string("double"));
   declare_vec<std::tuple<ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>>>(m, std::string("tuple_vec10"));
   declare_vec<std::tuple<ks::vec<double>, ks::vec<double>>>(m, std::string("tuple_vec2"));
diff --git a/test/ksc/gmmpy.cpp b/test/ksc/gmmpy.cpp
index 0d63763cb..ea0a4fe60 100644
--- a/test/ksc/gmmpy.cpp
+++ b/test/ksc/gmmpy.cpp
@@ -34,7 +34,7 @@ void declare_vec(py::module &m, std::string typestr) {
 // possible because Python tuples map to std::tuples regardless of
 // their contents.  I'll look into it later.  For now I'll just have a
 // bunch of verbose replication.
-PYBIND11_MODULE(MNISTCNNCPP_MODULE_NAME, m) {
+PYBIND11_MODULE(PYTHON_MODULE_NAME, m) {
   declare_vec<double>(m, std::string("double"));
   declare_vec<std::tuple<ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>, ks::vec<double>>>(m, std::string("tuple_vec10"));
   declare_vec<std::tuple<ks::vec<double>, ks::vec<double>>>(m, std::string("tuple_vec2"));
diff --git a/test/ksc/mnistcnnpy.cpp b/test/ksc/mnistcnnpy.cpp
index 7ece9fd93..c34eca558 100644
--- a/test/ksc/mnistcnnpy.cpp
+++ b/test/ksc/mnistcnnpy.cpp
@@ -24,7 +24,7 @@ void declare_vec(py::module &m, std::string typestr) {
 
 // In the future it might make more sense to move the vec type
 // definitions to a general Knossos CPP types Python module.
-PYBIND11_MODULE(MNISTCNNCPP_MODULE_NAME, m) {
+PYBIND11_MODULE(PYTHON_MODULE_NAME, m) {
   declare_vec<double>(m, std::string("double"));
   declare_vec<ks::vec<double> >(m, std::string("vec_double"));
   declare_vec<ks::vec<ks::vec<double> > >(m, std::string("vec_vec_double"));