Fixed unit test gotcha