Small fix on styling