diff --git a/Content/Scripts/upymodule.json b/Content/Scripts/upymodule.json
index a253a76..f666a56 100644
--- a/Content/Scripts/upymodule.json
+++ b/Content/Scripts/upymodule.json
@@ -1,6 +1,6 @@
 {
 	"name" : "tensorflow-ue4",
 	"pythonModules": {
-		"tensorflow": "1.1.0"
+		"tensorflow-gpu": "1.1.0"
 	}
 }
\ No newline at end of file