Fix failing unittest