param injection changed

Added injection at 2 places.