Fix margin issue