fixed display bug from before