diff --git a/python-docstring.el b/python-docstring.el
index 95ac740..e833066 100644
--- a/python-docstring.el
+++ b/python-docstring.el
@@ -32,7 +32,9 @@
 ;;; Code:
 
 (defvar python-docstring-script
-  (concat (file-name-as-directory (file-name-directory #$))
+  (concat (if load-file-name
+              (file-name-directory load-file-name)
+            default-directory)
           "docstring_wrap.py")
   "The location of the docstring_wrap.py script.")