Fixed diplay bug