Changed import statement in correct file