correst some display bugs