fix css
fix logic