fix import problem in ST2