Attempt to fix broken test