Fix bad display before drag