Moved import in proper position