fixed edit in controller