Fixed typo
Fixed import error and migration dependency.