diff --git a/python/src/test/java/org/apache/zeppelin/python/PythonInterpreterTest.java b/python/src/test/java/org/apache/zeppelin/python/PythonInterpreterTest.java
index dbd13462966..2e8060deec1 100644
--- a/python/src/test/java/org/apache/zeppelin/python/PythonInterpreterTest.java
+++ b/python/src/test/java/org/apache/zeppelin/python/PythonInterpreterTest.java
@@ -114,9 +114,9 @@ public void testPy4jIsNotInstalled() {
     assertNull(pythonInterpreter.getPy4jPort());
 
     assertTrue(cmdHistory.contains("def help()"));
-    assertTrue(cmdHistory.contains("class PyZeppelinContext():"));
+    assertTrue(cmdHistory.contains("class PyZeppelinContext(object):"));
     assertTrue(cmdHistory.contains("z = PyZeppelinContext"));
-    assertTrue(cmdHistory.contains("def zeppelin_show"));
+    assertTrue(cmdHistory.contains("def show"));
     assertFalse(cmdHistory.contains("GatewayClient"));
 
   }
@@ -140,9 +140,9 @@ public void testPy4jInstalled() {
     assertNotNull(py4jPort);
 
     assertTrue(cmdHistory.contains("def help()"));
-    assertTrue(cmdHistory.contains("class PyZeppelinContext():"));
-    assertTrue(cmdHistory.contains("z = PyZeppelinContext"));
-    assertTrue(cmdHistory.contains("def zeppelin_show"));
+    assertTrue(cmdHistory.contains("class PyZeppelinContext(object):"));
+    assertTrue(cmdHistory.contains("z = Py4jZeppelinContext"));
+    assertTrue(cmdHistory.contains("def show"));
     assertTrue(cmdHistory.contains("GatewayClient(port=" + py4jPort + ")"));
     assertTrue(cmdHistory.contains("org.apache.zeppelin.display.Input"));