Fix on modify button crash