Fix a small issue in the import logic that caused tests to fail.
Fix a small issue in the import logic that caused tests to fail.

tjwei/jedi,mfussenegger/jedi,flurischt/jedi,WoLpH/jedi,flurischt/jedi,jonashaag/jedi,mfussenegger/jedi,dwillmer/jedi,jonashaag/jedi,dwillmer/jedi,WoLpH/jedi,tjwei/jedi