Move the ImportError to the right location