diff --git a/example/example_live_test.go b/example/example_live_test.go
index d2bbaac..99d0f0e 100644
--- a/example/example_live_test.go
+++ b/example/example_live_test.go
@@ -1,4 +1,4 @@
-package example
+package main
 
 import (
 	"log/slog"
@@ -74,7 +74,6 @@ func (s *MySuite) TestLightService() {
 	s.app.GetService().Light.Toggle(entityId)
 
 	time.Sleep(1 * time.Second) // wait for state to update
-	// <-time.After(1 * time.Second)
 
 	newState, err := s.app.GetState().Get(entityId)
 	if err != nil {
diff --git a/example/go.mod b/example/go.mod
index 95813bf..3121021 100644
--- a/example/go.mod
+++ b/example/go.mod
@@ -1,4 +1,4 @@
-module example
+module main
 
 go 1.21.0