diff --git a/cli_bilby/resume.py b/cli_bilby/resume.py
index e035aedc40f27d1de5dfa43994770f570aec1983..e808c5febd896d9100105d2d6d0dc6e314341e63 100644
--- a/cli_bilby/resume.py
+++ b/cli_bilby/resume.py
@@ -78,7 +78,7 @@ def convert_df_to_preresult(df, format, resume_file):
     result.posterior = df
     result.priors = dict()
     filename = bilby.core.result.result_file_name(outdir, label, format)
-    filename = filename.replace("result", "preresult")
+    filename = filename.replace("result.{}".format(format), "preresult.{}".format(format))
     result.save_to_file(filename=filename, extension=format)
 
 
diff --git a/setup.py b/setup.py
index 331d7f8d08238264ff626e45d5e38a54a5881174..9edd5b0204f99a83f10fc6cb9038dbeef96462e3 100644
--- a/setup.py
+++ b/setup.py
@@ -72,7 +72,7 @@ setup(name='bilby',
       packages=['bilby', 'bilby.core', 'bilby.core.sampler',
                 'bilby.gw', 'bilby.gw.detector', 'bilby.gw.sampler',
                 'bilby.hyper', 'cli_bilby'],
-      package_dir={'bilby': 'bilby'},
+      package_dir={'bilby': 'bilby', 'cli_bilby': 'cli_bilby'},
       package_data={'bilby.gw': ['prior_files/*'],
                     'bilby.gw.detector': ['noise_curves/*.txt', 'detectors/*'],
                     'bilby': [version_file]},