fix the margin problem