diff --git a/RMtools_3D/mk_test_cube_data.py b/RMtools_3D/mk_test_cube_data.py
index 81acf2b..d0f8457 100644
--- a/RMtools_3D/mk_test_cube_data.py
+++ b/RMtools_3D/mk_test_cube_data.py
@@ -420,7 +420,7 @@ def create_IQU_cube_data(
 
     # Loop through the frequency channels & insert the IQU planes
     for iChan, _ in enumerate(
-        tqdm(freqArr_Hz), desc=f"Looping through {nChans} frequency channels"
+        tqdm(freqArr_Hz, desc=f"Looping through {nChans} frequency channels")
     ):
         for iSrc, _ in enumerate(spectraILst):
             params = [
diff --git a/setup.py b/setup.py
index 627d8c5..910ace6 100644
--- a/setup.py
+++ b/setup.py
@@ -70,7 +70,7 @@
             "rmtools_extractregion=RMtools_3D.extract_region:main",
             "rmtools_bwdepol=RMtools_1D.rmtools_bwdepol:main",
             "rmtools_bwpredict=RMtools_1D.rmtools_bwpredict:main",
-            "rmtools_3DIrescale=RMtools_3D.rescale_I_model_3D:main",
+            "rmtools_3DIrescale=RMtools_3D.rescale_I_model_3D:command_line",
         ],
     },
     install_requires=REQUIRED,