Fix another problem with tests failing on travis