fix ut import err