diff --git a/examples/nat-gateway/main.tf b/examples/nat-gateway/main.tf
index 6c8c58d..7a6e5e9 100644
--- a/examples/nat-gateway/main.tf
+++ b/examples/nat-gateway/main.tf
@@ -4,8 +4,8 @@ locals {
   firezone_token = "YOUR_FIREZONE_TOKEN"
 }
 
-module "aws_firezone_gateway" {
-  source = "github.com/firezone/terraform-firezone-aws"
+module "gateway" {
+  source = "firezone/gateway/aws"
 
   ###################
   # Required inputs #