make test code actually works