Fix modal bug