Fixed old version of unit test