Fixed a unittest