Add inject support