Made test 2 pass