Build fix for Travis CI