Updated code to remove Travis errors.