diff --git a/main.tftest.hcl b/main.tftest.hcl
index 68f4c3b..88f63e1 100644
--- a/main.tftest.hcl
+++ b/main.tftest.hcl
@@ -5,11 +5,10 @@ provider "aws" {
 
 run "main" {
   assert {
-    condition     = output.arn != null
+    condition     = module.main.arn != null
     error_message = "invalid IAM role ARN"
   }
-}
-
-variables {
-  name = "tftest-role"
+  module {
+    source = "./test"
+  }
 }