Make edit draft work