diff --git a/halo/halo.py b/halo/halo.py
index cab2185..e25b250 100644
--- a/halo/halo.py
+++ b/halo/halo.py
@@ -99,7 +99,7 @@ def __init__(
 
         environment = get_environment()
 
-        def clean_up():
+        def clean_up(args):
             """Handle cell execution"""
             self.stop()
 
diff --git a/halo/halo_notebook.py b/halo/halo_notebook.py
index d2c8c91..3bf097b 100644
--- a/halo/halo_notebook.py
+++ b/halo/halo_notebook.py
@@ -53,6 +53,7 @@ def clear(self):
             self.output.outputs += self._output(self.CLEAR_LINE)
 
         self.output.outputs = self._output()
+        self.output.close()
         return self
 
     def _render_frame(self):