add fix for test