Fix a import error