Fix failing test-case