diff --git a/optimus/df_analyzer.py b/optimus/df_analyzer.py
index f26f9e7a9..fad0147a9 100644
--- a/optimus/df_analyzer.py
+++ b/optimus/df_analyzer.py
@@ -817,8 +817,7 @@ def write_json(cls, json_cols, path_to_json_file):
             # outfile.write(str(json_cols).replace("'", "\""))
             outfile.write(json_cols)
 
-    @classmethod
-    def display_optimus(cls, df):
+    def display_optimus(self):
         """
         Lets you visualize your Spark object in different ways: table, charts, maps, etc.
         :param df: spark dataframe to be analyzed
@@ -826,7 +825,7 @@ def display_optimus(cls, df):
         """
         # Import pixiedust-optimus
         from pixiedust_optimus.display import display
-        display(df)
+        display(self._df)
 
     def get_frequency(self, columns, sort_by_count=True):
         """