Fix test, this will make the travis build pass