fixed bug with display