Add additional parameter to better handle retrieve for edit.