Updated test to pass