fixed it correctly in controller