1 files changed, 2 insertions, 2 deletions
diff --git a/setup.py b/setup.py
index 359540d..ff8a420 100644
--- a/setup.py
+++ b/setup.py
@@ -10,9 +10,9 @@ all_extensions = Extension("*", ["pyisda/*.pyx"],
                            extra_compile_args=['-fopenmp'],
                            extra_link_args=['-fopenmp', '-Wl,--strip-all'])
 
-c_extension = Extension("pyisda.flat_hazard",
+c_extension = Extension("pyisda.optim",
                         include_dirs=['c_layer', numpy.get_include()],
-                        sources=['pyisda/flat_hazard.pyx', 'c_layer/cdsbootstrap.c'],
+                        sources=['pyisda/optim.pyx', 'c_layer/cdsbootstrap.c'],
                         libraries=['cds'],
                         language='c++')