diff --git a/demo/src/gui.py b/demo/src/gui.py
index e12b2de..62609ff 100644
--- a/demo/src/gui.py
+++ b/demo/src/gui.py
@@ -154,7 +154,9 @@ def run(self):
                         )
 
                         self.slider.input(
-                            self.get_img_pred_pair, self.slider, t,
+                            self.get_img_pred_pair,
+                            self.slider,
+                            t,
                         )
 
                         self.slider.render()
diff --git a/demo/src/utils.py b/demo/src/utils.py
index b05aa4b..c6df5ba 100644
--- a/demo/src/utils.py
+++ b/demo/src/utils.py
@@ -5,7 +5,7 @@
 
 
 def load_ct_to_numpy(data_path):
-    if type(data_path) != str:
+    if not isinstance(data_path, str):
         data_path = data_path.name
 
     image = nib.load(data_path)
@@ -25,7 +25,7 @@ def load_ct_to_numpy(data_path):
 
 
 def load_pred_volume_to_numpy(data_path):
-    if type(data_path) != str:
+    if not isinstance(data_path, str):
         data_path = data_path.name
 
     image = nib.load(data_path)