diff --git a/.gitignore b/.gitignore
index b7347fc..7a0a67c 100644
--- a/.gitignore
+++ b/.gitignore
@@ -3,6 +3,7 @@ upsonic_cache/
 .vscode
 
 test.py
+test.ipynb
 
 test_backup
 *.Upsonic
diff --git a/upsonic/remote/on_prem.py b/upsonic/remote/on_prem.py
index bdd429d..2da75ac 100644
--- a/upsonic/remote/on_prem.py
+++ b/upsonic/remote/on_prem.py
@@ -815,12 +815,7 @@ def set(
                 self._log(f"Error on extract_source while dumping {key}")
                 traceback.print_exc()
 
-        try:
-            the_engine_reports["extract_source"] = fernet.encrypt(pickle.dumps(extract_source(value, self.tester), protocol=1))
-        except:
-            if self.tester:
-                self._log(f"Error on extract_source while dumping {key}")
-                traceback.print_exc()
+
 
         if extracted_needed_libraries != None:
             the_engine_reports["extract_needed_libraries"] = fernet.encrypt(pickle.dumps(extracted_needed_libraries, protocol=1))
@@ -844,6 +839,10 @@ def set(
         else:
             return False
 
+
+    def print_code(self, key, version=None):
+        print(self.get(key, version=version, extract_source=True))
+
     def get(
             self,
             key,
@@ -851,7 +850,8 @@ def get(
             print_exc=True,
             pass_python_version_control=False,
             pass_usage_analyses=False,
-            try_to_extract_importable=False
+            try_to_extract_importable=False,
+            extract_source=False
 
     ):
         if self.tester:
@@ -928,6 +928,8 @@ def get(
                         self._log(f"Error on extracted_local_files while loading {key}")
                         traceback.print_exc()
                 response.pop("extracted_local_files")
+            if extract_source:
+                return pickle.loads(fernet.decrypt(response["extract_source"]))
             if "extract_source" in response:
                 response.pop("extract_source")
             needed_libraries = None