modifying test to reflect code change