Fix for travis build