Added fix for travis build.