Capturing the original state properly in the controller middleware