Fix: make test pass