fix tests

Passing on travis again.