UI fix for modal sizes