Unittest fixed