fix for test