modified test to pass