diff --git a/custom-domain/dstack-ingress/scripts/certman.py b/custom-domain/dstack-ingress/scripts/certman.py
index 68809e8..2c037d3 100644
--- a/custom-domain/dstack-ingress/scripts/certman.py
+++ b/custom-domain/dstack-ingress/scripts/certman.py
@@ -10,7 +10,7 @@
 # Add script directory to path to import dns_providers
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
 
-from .dns_providers.base import DNSProvider
+
 from dns_providers import DNSProviderFactory
 
 
diff --git a/custom-domain/dstack-ingress/scripts/entrypoint.sh b/custom-domain/dstack-ingress/scripts/entrypoint.sh
index a008b33..724469f 100644
--- a/custom-domain/dstack-ingress/scripts/entrypoint.sh
+++ b/custom-domain/dstack-ingress/scripts/entrypoint.sh
@@ -3,7 +3,11 @@
 set -e
 
 PORT=${PORT:-443}
-TXT_PREFIX=${TXT_PREFIX:-"_tapp-address"}
+if [[ -e /var/run/dstack.sock ]]; then
+  TXT_PREFIX=${TXT_PREFIX:-"_dstack-app-address"}
+else
+  TXT_PREFIX=${TXT_PREFIX:-"_tapp-address"}
+fi
 
 echo "Setting up certbot environment"