diff --git a/examples/basic/basic/outputs.tf b/examples/basic/basic/outputs.tf
index 23b47a7..8a9938c 100644
--- a/examples/basic/basic/outputs.tf
+++ b/examples/basic/basic/outputs.tf
@@ -1,3 +1,4 @@
 output "user_data" {
-  value = module.TestBasic.user_data
+  value     = module.TestBasic.user_data
+  sensitive = true
 }
\ No newline at end of file
diff --git a/tests/util_test.go b/tests/util_test.go
index 49b9f81..52ee16e 100644
--- a/tests/util_test.go
+++ b/tests/util_test.go
@@ -16,6 +16,13 @@ import (
 func teardown(t *testing.T, category string, directory string, keyPair *aws.Ec2Keypair) {
 	err := os.RemoveAll(fmt.Sprintf("../examples/%s/%s/.terraform", category, directory))
 	require.NoError(t, err)
+	err1 := os.RemoveAll(fmt.Sprintf("../examples/%s/%s/.terraform.lock.hcl", category, directory))
+	require.NoError(t, err1)
+	err2 := os.RemoveAll(fmt.Sprintf("../examples/%s/%s/terraform.tfstate", category, directory))
+	require.NoError(t, err2)
+	err3 := os.RemoveAll(fmt.Sprintf("../examples/%s/%s/terraform.tfstate.backup", category, directory))
+	require.NoError(t, err3)
+
 	aws.DeleteEC2KeyPair(t, keyPair)
 }