Change some asserts which weren't working on travis