Bug fix for travis failure