Fix margin problem