Fixing to work on travis