Fixed some minor import issues.