Fix imcompatible import statement