small fix for test