diff --git a/unittests/test_pi_import.py b/unittests/test_pi_import.py
index e5bb02db..765bf396 100644
--- a/unittests/test_pi_import.py
+++ b/unittests/test_pi_import.py
@@ -1,5 +1,6 @@
 import unittest
 import sys, os, subprocess
+import wx
 
 #---------------------------------------------------------------------------
 
@@ -16,7 +17,7 @@ class PIImportTest(unittest.TestCase):
 
     def runPI(self, filename):
         cwd = os.getcwd()
-        dirname = os.path.join(os.path.dirname(__file__), '../wx')
+        dirname = os.path.dirname(wx.__file__)
         os.chdir(dirname)
 
         sp = subprocess.Popen('%s %s' % (sys.executable, filename), 
@@ -66,4 +67,4 @@ class PIImportTest(unittest.TestCase):
 
 if __name__ == '__main__':
     unittest.main()
-        
\ No newline at end of file
+