diff --git a/pkg/discovery/injector_test.go b/pkg/discovery/injector_test.go
index 2b99bbc1..8e70bf75 100644
--- a/pkg/discovery/injector_test.go
+++ b/pkg/discovery/injector_test.go
@@ -492,6 +492,14 @@ func TestDefaultInjector_InjectEnv(t *testing.T) {
 					Name:  "RBG_GROUP_NAME",
 					Value: "test-rbg",
 				},
+				{
+					Name: "RBG_ROLE_INDEX",
+					ValueFrom: &corev1.EnvVarSource{
+						FieldRef: &corev1.ObjectFieldSelector{
+							FieldPath: "metadata.labels['apps.kubernetes.io/pod-index']",
+						},
+					},
+				},
 				{
 					Name:  "RBG_ROLE_NAME",
 					Value: "worker",
@@ -551,6 +559,14 @@ func TestDefaultInjector_InjectEnv(t *testing.T) {
 					Name:  "RBG_GROUP_NAME",
 					Value: "test-rbg",
 				},
+				{
+					Name: "RBG_ROLE_INDEX",
+					ValueFrom: &corev1.EnvVarSource{
+						FieldRef: &corev1.ObjectFieldSelector{
+							FieldPath: "metadata.labels['apps.kubernetes.io/pod-index']",
+						},
+					},
+				},
 				{
 					Name:  "RBG_ROLE_NAME",
 					Value: "worker",