diff --git a/src/sql/parts/notebook/notebookInput.ts b/src/sql/parts/notebook/notebookInput.ts
index 196decebb6..262afe02ed 100644
--- a/src/sql/parts/notebook/notebookInput.ts
+++ b/src/sql/parts/notebook/notebookInput.ts
@@ -164,6 +164,10 @@ export class NotebookInput extends EditorInput {
 		return this._model.confirmSave();
 	}
 
+	public revert(): TPromise<boolean> {
+		return this._textInput.revert();
+	}
+
 	public get notebookUri(): URI {
 		return this.resource;
 	}