diff --git a/CMakeLists.txt b/CMakeLists.txt
index 1958fd7..2359669 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -16,7 +16,7 @@ set(CMAKE_CXX_STANDARD_REQUIRED True)
 if (WITH_PYTHON)
 add_subdirectory(vendor/pybind11)
 add_definitions("-DVERSION_INFO=${VERSION_INFO}")
-pybind11_add_module(fastshuffle fastshuffle.cpp)
+pybind11_add_module(fastshuffle fastshuffle.cpp shuffle.cpp)
 install(TARGETS fastshuffle LIBRARY DESTINATION lib)
 endif()
 
diff --git a/fastshuffle.cpp b/fastshuffle.cpp
index 0effeec..0ee8ad6 100644
--- a/fastshuffle.cpp
+++ b/fastshuffle.cpp
@@ -6,7 +6,7 @@
 
 namespace py = pybind11;
 
-PYBIND11_MODULE(fastmerge, m) {
+PYBIND11_MODULE(fastshuffle, m) {
     m.def("file_shuffle", &shuffle);
 
 #ifdef VERSION_INFO