Fix bug picked up by Travis