Fixing import typo