Fixes the tests in 0.8

This should fox the travis build.