diff --git a/charts/netbox/templates/ConfigMap/netbox.yaml b/charts/netbox/templates/ConfigMap/netbox.yaml
index 82d2cbc6..e90e7880 100644
--- a/charts/netbox/templates/ConfigMap/netbox.yaml
+++ b/charts/netbox/templates/ConfigMap/netbox.yaml
@@ -8,6 +8,9 @@ metadata:
   labels: {{- include "common.labels.standard" (dict "customLabels" .Values.commonLabels "context" $) | nindent 4 }}
 data:
   configuration.py: |-
+    {{- if .Values.allowedHostsIncludesPodIp }}
+    import os
+    {{- end }}
     import re
     from pathlib import Path