Correct import mistake