Adding fix for Travis.ci
Adding fix for ci