Added fix for travis