Now the test code works :D