Corrected a bug failing travis test