diff --git a/src/unitgrade_devel.egg-info/PKG-INFO b/src/unitgrade_devel.egg-info/PKG-INFO
index 33c1164e1b9d1de8155d7fa799f2fabbb7ccc03a..cc0a865d0a1bd5058de229b19bd9b1da6b2defd3 100644
--- a/src/unitgrade_devel.egg-info/PKG-INFO
+++ b/src/unitgrade_devel.egg-info/PKG-INFO
@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unitgrade-devel
-Version: 0.1.61
+Version: 0.1.62
 Summary: A set of tools to develop unitgrade tests and reports and later evaluate them
 Home-page: https://lab.compute.dtu.dk/tuhe/unitgrade_private
 Author: Tue Herlau
diff --git a/src/unitgrade_private/deployment.py b/src/unitgrade_private/deployment.py
index e228e7dc9f1506b0e5e00c799cebc316cdefad6c..93756bf3ac582a1b2d302c451426e8dfd6751186 100644
--- a/src/unitgrade_private/deployment.py
+++ b/src/unitgrade_private/deployment.py
@@ -74,6 +74,7 @@ def remove_hidden_methods(ReportClass, outfile=None):
     # Allows us to use the !b;silent tags in the code. This is a bit hacky, but allows timeouts, etc. to make certain tests more robust
     from snipper.fix_bf import fix_b
     from snipper.snipper_main import fix_tags
+    from snipper.snip_dir import censor_file
     from snipper.snip_dir import snip_dir
 
     lines, _, _ = fix_b(fix_tags(source.rstrip().splitlines()))