Fix that import mistake