Fixed bug noticed by Travis CI build