Replace layoutSubviews with layoutIfNeeded