Fix Edit page bug