Fix failing spec