modify test to pass