Small fix for modals