Another incorrect import fixed