trying to fix UT bug