diff --git a/jupyter_rstudio_singularity/cli.py b/jupyter_rstudio_singularity/cli.py
index b586cfd..fbd96a5 100644
--- a/jupyter_rstudio_singularity/cli.py
+++ b/jupyter_rstudio_singularity/cli.py
@@ -7,6 +7,8 @@
 from contextlib import closing
 
 from prefect import flow, task
+from prefect import flow
+from prefect_shell import shell_run_command
 import os
 import tempfile
 from jupyter_rstudio_singularity.utils.logging import setup_logger