diff --git a/erpnext_germany/erpnext_germany/doctype/vat_id_check/vat_id_check.js b/erpnext_germany/erpnext_germany/doctype/vat_id_check/vat_id_check.js
index 1c272d6..7604ba8 100644
--- a/erpnext_germany/erpnext_germany/doctype/vat_id_check/vat_id_check.js
+++ b/erpnext_germany/erpnext_germany/doctype/vat_id_check/vat_id_check.js
@@ -1,8 +1,16 @@
 // Copyright (c) 2023, ALYF GmbH and contributors
 // For license information, please see license.txt
 
-frappe.ui.form.on('VAT ID Check', {
-	// refresh: function(frm) {
-
-	// }
+frappe.ui.form.on("VAT ID Check", {
+	setup: function (frm) {
+		frm.set_query("customer_address", function (doc) {
+			return {
+				query: "frappe.contacts.doctype.address.address.address_query",
+				filters: {
+					link_doctype: "Customer",
+					link_name: doc.customer,
+				},
+			};
+		});
+	},
 });