Fix typo
Added missing import