Adding missing parameter to UT