Hopefully done the wrap in controller call correctly.