Correcting tests to pass on travis