Trying to get current injection correct