Make small change in unit test