Fixed generated test to pass