Move import to correct file