Added option to inject connection