Fix another unittest error