Fix for travis tests