fixed problem with display