diff --git a/python/ipnb/src/org/jetbrains/plugins/ipnb/psi/IpnbPyFragment.java b/python/ipnb/src/org/jetbrains/plugins/ipnb/psi/IpnbPyFragment.java
index f4f29e75ec91..0ef5ccfbf46a 100644
--- a/python/ipnb/src/org/jetbrains/plugins/ipnb/psi/IpnbPyFragment.java
+++ b/python/ipnb/src/org/jetbrains/plugins/ipnb/psi/IpnbPyFragment.java
@@ -77,7 +77,7 @@ public class IpnbPyFragment extends PyFileImpl {
 
   @NotNull
   @Override
-  public PsiFile getOriginalFile() {
+  public PsiFile getContext() {
     final PsiFile psiFile = PsiDocumentManager.getInstance(getProject()).getPsiFile(myFilePanel.getDocument());
     return psiFile != null ? psiFile : super.getOriginalFile();
   }