diff --git a/zeppelin-zengine/src/test/java/org/apache/zeppelin/notebook/NoteTest.java b/zeppelin-zengine/src/test/java/org/apache/zeppelin/notebook/NoteTest.java
index 1a1ea162494..fda4e4e3249 100644
--- a/zeppelin-zengine/src/test/java/org/apache/zeppelin/notebook/NoteTest.java
+++ b/zeppelin-zengine/src/test/java/org/apache/zeppelin/notebook/NoteTest.java
@@ -139,7 +139,7 @@ public void addParagraphWithLastReplName() {
 
     Note note = new Note(repo, interpreterFactory, jobListenerFactory, index, credentials, noteEventListener);
     note.putDefaultReplName(); //set lastReplName
-    when(note.isBinding("spark")).thenReturn(true);
+    when(interpreterFactory.getInterpreter(note.getId(), "spark")).thenReturn(new MockInterpreter2(null));
 
     Paragraph p = note.addParagraph();
 
@@ -155,7 +155,7 @@ public void insertParagraphWithLastReplName() {
 
     Note note = new Note(repo, interpreterFactory, jobListenerFactory, index, credentials, noteEventListener);
     note.putDefaultReplName(); //set lastReplName
-    when(note.isBinding("spark")).thenReturn(true);
+    when(interpreterFactory.getInterpreter(note.getId(), "spark")).thenReturn(new MockInterpreter2(null));
 
     Paragraph p = note.insertParagraph(note.getParagraphs().size());