Typo
Fix some changed imports #72