Fix testing

This should make the TravisCI tests pass