add proper injection