diff --git a/webshell/tests/viewer/nsEditorMode.cpp b/webshell/tests/viewer/nsEditorMode.cpp
index 22499cd9b2b..bf36fcec2ad 100644
--- a/webshell/tests/viewer/nsEditorMode.cpp
+++ b/webshell/tests/viewer/nsEditorMode.cpp
@@ -102,9 +102,7 @@ nsresult NS_InitEditorMode(nsIDOMDocument *aDOMDocument, nsIPresShell* aPresShel
     return NS_ERROR_OUT_OF_MEMORY;
   }
 
-  gEditor->InitHTMLEditor(aDOMDocument, aPresShell);
-  gEditor->EnableUndo(PR_TRUE);
-
+  result = gEditor->Init(aDOMDocument, aPresShell);
   return result;
 }