fixed bug in test that didn't work on travis