Fixed a glitch in unit tests