Fixed edit assignment button missing bug