Bug fix + import error