diff --git a/grott/Dockerfile b/grott/Dockerfile
index 1bd3857..a0418a0 100644
--- a/grott/Dockerfile
+++ b/grott/Dockerfile
@@ -23,7 +23,8 @@ RUN pip3 install --no-cache --upgrade pip setuptools
 COPY script.sh /app/script.sh
 WORKDIR /app
 RUN curl -Lk 'https://github.com/johanmeijer/grott/archive/refs/heads/2.7-(Beta).zip' --output files.zip && unzip -o files.zip && rm files.zip && mv grott-*/*.py ./
-RUN curl -L "https://raw.githubusercontent.com/egguy/grott/feature/ha-extension/examples/Home%20Assistent/grott_ha.py" -o grott_ha.py
+# Move the addon
+RUN mv grott-*/examples/Home\ Assistent/grott_ha.py ./
 
 #Install required python packages 
 RUN pip install paho-mqtt requests influxdb influxdb
diff --git a/grott/config.json b/grott/config.json
index 4f65e00..55406e3 100644
--- a/grott/config.json
+++ b/grott/config.json
@@ -1,6 +1,6 @@
 {
-	"name": "Grott",
-	"version": "2.7.6",
+	"name": "Grott - HA extension",
+	"version": "2.7.7",
 	"slug": "grott",
 	"description": "The Growatt inverter monitor",
 	"url": "https://github.com/johanmeijer/grott",
@@ -24,6 +24,7 @@
 	"options": {
 		"gmode": "proxy",
 		"gnomqtt": "False",
+		"ginvtype": "default",
 		"mqtt": {}
 	},
 	"schema": {
@@ -69,6 +70,7 @@
 		"gextension": "bool?",
 		"gextname": "str?",
 		"gextvar": "str?",
+		"ginvtypemap": "str?",
 		"mqtt": {
 			"server": "str?",
 			"user": "str?",
diff --git a/grott/script.sh b/grott/script.sh
index bdb8c43..07ed0f1 100644
--- a/grott/script.sh
+++ b/grott/script.sh
@@ -39,6 +39,8 @@ if  bashio::config.has_value 'gifpassword'; then export gifpassword="$(bashio::c
 if  bashio::config.has_value 'giforg'; then export giforg="$(bashio::config 'giforg')"; fi
 if  bashio::config.has_value 'gifbucket'; then export gifbucket="$(bashio::config 'gifbucket')"; fi
 if  bashio::config.has_value 'giftoken'; then export giftoken="$(bashio::config 'giftoken')"; fi
+if  bashio::config.has_value 'ginvtypemap'; then export ginvtypemap="$(bashio::config 'ginvtypemap')"; fi
+
 
 export gextension="True"
 export gextname="grott_ha"