Fix tests

Lets see if Travis runs too, it doesn't seem like it was working.