on small tweak to unit test