diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index fa00db9..937163a 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -193,20 +193,22 @@ jobs:
               "github.com/user_account/bose-soundtouch/pkg/client"
               "github.com/user_account/bose-soundtouch/pkg/models"
               "github.com/user_account/bose-soundtouch/pkg/discovery"
+              "github.com/user_account/bose-soundtouch/pkg/config"
           )
 
           func main() {
               // Test basic client creation
-              c := client.New("192.168.1.100", 8090)
+              c := client.NewClientFromHost("192.168.1.100")
               fmt.Printf("Client created for %s\n", c.BaseURL())
 
               // Test models can be imported
-              var info models.Info
-              fmt.Printf("Info model available: %T\n", info)
+              var info models.DeviceInfo
+              fmt.Printf("DeviceInfo model available: %T\n", info)
 
               // Test discovery can be imported
-              var scanner discovery.Scanner
-              fmt.Printf("Scanner available: %T\n", scanner)
+              cfg := config.DefaultConfig()
+              service := discovery.NewUnifiedDiscoveryService(cfg)
+              fmt.Printf("Discovery service available: %T\n", service)
 
               fmt.Println("All imports successful!")
           }