Fixing Travis tests