Add missing argument in setUserState