Fixed import in the correct file