Fixed trivial bug in unit tests