Fixed regressed unit test