Fix tests for flat_dialog_mod