Fix typo & import error #2