fix silly import error