Fix failing test #2