Needed to fix issue with controller in api change.