Fixing styling bug