TODO: fix css bug