diff --git a/python/src/com/jetbrains/python/testing/PythonTestConfigurationProducer.java b/python/src/com/jetbrains/python/testing/PythonTestConfigurationProducer.java
index 4b3085cf5d5d..06433551fa6f 100644
--- a/python/src/com/jetbrains/python/testing/PythonTestConfigurationProducer.java
+++ b/python/src/com/jetbrains/python/testing/PythonTestConfigurationProducer.java
@@ -33,10 +33,7 @@ import com.intellij.openapi.roots.ProjectRootManager;
 import com.intellij.openapi.util.Ref;
 import com.intellij.openapi.util.text.StringUtil;
 import com.intellij.openapi.vfs.VirtualFile;
-import com.intellij.psi.PsiDirectory;
-import com.intellij.psi.PsiElement;
-import com.intellij.psi.PsiFile;
-import com.intellij.psi.PsiFileSystemItem;
+import com.intellij.psi.*;
 import com.intellij.psi.util.PsiTreeUtil;
 import com.jetbrains.python.PythonModuleTypeBase;
 import com.jetbrains.python.facet.PythonFacetSettings;
@@ -116,7 +113,10 @@ abstract public class PythonTestConfigurationProducer extends RunConfigurationPr
     if (context == null) return false;
     final Location location = context.getLocation();
     if (location == null || !isAvailable(location)) return false;
-    final PsiElement element = location.getPsiElement();
+    PsiElement element = location.getPsiElement();
+    if (element instanceof PsiWhiteSpace) {
+      element = PyUtil.findNonWhitespaceAtOffset(element.getContainingFile(), element.getTextOffset());
+    }
 
     if (PythonUnitTestRunnableScriptFilter.isIfNameMain(location)) return false;
     final Module module = location.getModule();