small import code change