Fix failing Unittest