diff --git a/active_plugins/runcellpose.py b/active_plugins/runcellpose.py
index 47426d2..6edc915 100644
--- a/active_plugins/runcellpose.py
+++ b/active_plugins/runcellpose.py
@@ -601,10 +601,13 @@ def run(self, workspace):
             # Save the image to the Docker mounted directory
             skimage.io.imsave(temp_img_path, x_data)
 
-            cmd = [docker_path, 'run', '--rm', '-v', f'{temp_dir}:/data', self.docker_image.value]
+            cmd = [docker_path, 'run', '--rm', '-v', f'{temp_dir}:/data',]
             if self.use_gpu.value:
                 cmd += ['--gpus', 'all']
-            cmd += ['cellpose', '--verbose', '--dir', '/data/img', '--pretrained_model']
+            cmd += [self.docker_image.value, 'cellpose']
+            if self.use_gpu.value:
+                cmd += ['--use_gpu']
+            cmd += ['--verbose', '--dir', '/data/img', '--pretrained_model']
             if self.mode.value !='custom':
                 cmd += [self.mode.value]
             else: