FIX broken test