Actually fix broken test