corrected a bug with display