Fixed import error when executing run.py