Fix typo/incorrect name
Minor fixup to imports