Correct import and format slightly