Fix failing test: