Fix ImportError
Merge pull request #1 from matuu/feature/FixImportErrorEditModule

Fix ImportError