Fix element passed to the revertAction function