Minor tweak to make write unit test easier