Fix error message
Organize import