diff --git a/picasso/visualizations/partial_occlusion.py b/picasso/visualizations/partial_occlusion.py
index <HASH>..<HASH> 100644
--- a/picasso/visualizations/partial_occlusion.py
+++ b/picasso/visualizations/partial_occlusion.py
@@ -34,7 +34,7 @@ class PartialOcclusion(BaseVisualization):
     }
 
     def __init__(self, model):
-        super(PartialOcclusion, self).__init__(model)
+        super().__init__(model)
         self.predict_tensor = self.get_predict_tensor()
 
         self.window = 0.10
diff --git a/picasso/visualizations/saliency_maps.py b/picasso/visualizations/saliency_maps.py
index <HASH>..<HASH> 100644
--- a/picasso/visualizations/saliency_maps.py
+++ b/picasso/visualizations/saliency_maps.py
@@ -27,7 +27,7 @@ class SaliencyMaps(BaseVisualization):
     REFERENCE_LINK = 'https://arxiv.org/pdf/1312.6034'
 
     def __init__(self, model, logit_tensor_name=None):
-        super(SaliencyMaps, self).__init__(model)
+        super().__init__(model)
         if logit_tensor_name:
             self.logit_tensor = self.model.sess.graph.get_tensor_by_name(
                 logit_tensor_name)
make super calls python3ish (#<I>)