trying to fix broken test