Fix failing test setup