update to work with optimizations in mojito@0.5.0

I wasn't sure how to run the tests :(