Fixed test to match latest changes

refs 34d308c5070384bdb483c59da06ae11510caa397