diff --git a/src/test/java/com/embabel/impromptu/TestDataHelper.java b/src/test/java/com/embabel/impromptu/TestDataHelper.java
index 3d0079d..bfab1cc 100644
--- a/src/test/java/com/embabel/impromptu/TestDataHelper.java
+++ b/src/test/java/com/embabel/impromptu/TestDataHelper.java
@@ -235,7 +235,7 @@ public void clearTestData() {
      */
     public long countComposers() {
         return persistenceManager.getOne(
-                QuerySpecification.withStatement("MATCH (c:Composer) RETURN count(c)")
+                QuerySpecification.withStatement("MATCH (c:Composer) WHERE c.primarySource = 'test' RETURN count(c)")
                         .transform(Long.class)
         );
     }
@@ -245,7 +245,7 @@ public long countComposers() {
      */
     public long countWorks() {
         return persistenceManager.getOne(
-                QuerySpecification.withStatement("MATCH (w:Work) RETURN count(w)")
+                QuerySpecification.withStatement("MATCH (w:Work) WHERE w.primarySource = 'test' RETURN count(w)")
                         .transform(Long.class)
         );
     }
diff --git a/src/test/java/com/embabel/impromptu/TestLlmConfiguration.java b/src/test/java/com/embabel/impromptu/TestLlmConfiguration.java
index 3ccffb5..3fd20ce 100644
--- a/src/test/java/com/embabel/impromptu/TestLlmConfiguration.java
+++ b/src/test/java/com/embabel/impromptu/TestLlmConfiguration.java
@@ -127,26 +127,11 @@ public EmbeddingResponse call(EmbeddingRequest request) {
             return new EmbeddingResponse(List.of());
         }
 
-        @Override
-        public float[] embed(String text) {
-            return new float[1536];
-        }
-
         @Override
         public float[] embed(Document document) {
             return new float[1536];
         }
 
-        @Override
-        public List<float[]> embed(List<String> texts) {
-            return texts.stream().map(t -> new float[1536]).toList();
-        }
-
-        @Override
-        public EmbeddingResponse embedForResponse(List<String> texts) {
-            return new EmbeddingResponse(List.of());
-        }
-
         @Override
         public int dimensions() {
             return 1536;
@@ -157,7 +142,6 @@ public int dimensions() {
      * A simple fake EmbeddingService for tests.
      */
     static class FakeEmbeddingService implements EmbeddingService {
-        private final FakeEmbeddingModel model = new FakeEmbeddingModel();
 
         @Override
         public String getName() {
@@ -186,11 +170,6 @@ public List<float[]> embed(List<String> texts) {
         public int getDimensions() {
             return 1536;
         }
-
-        @Override
-        public Object getModel() {
-            return model;
-        }
     }
 
     /**
diff --git a/src/test/java/com/embabel/impromptu/integrations/youtube/YouTubePerformancePersistenceIntegrationTest.java b/src/test/java/com/embabel/impromptu/integrations/youtube/YouTubePerformancePersistenceIntegrationTest.java
index 80b7c83..be388bf 100644
--- a/src/test/java/com/embabel/impromptu/integrations/youtube/YouTubePerformancePersistenceIntegrationTest.java
+++ b/src/test/java/com/embabel/impromptu/integrations/youtube/YouTubePerformancePersistenceIntegrationTest.java
@@ -65,7 +65,7 @@ class YouTubePerformancePersistenceIntegrationTest {
     private NamedEntityDataRepository namedEntityDataRepository;
 
     private static final String TEST_VIDEO_ID = "dQw4w9WgXcQ";
-    private static final String TEST_PERFORMANCE_ID = TEST_VIDEO_ID;
+    private static final String TEST_PERFORMANCE_ID = "perf-" + TEST_VIDEO_ID;
     private static final String TEST_WORK_ID = "work-brahms-symphony4";
 
     @BeforeEach
@@ -123,7 +123,7 @@ private void createTestData() {
                             perf.performer = $performer,
                             perf.ensemble = $ensemble,
                             perf.conductor = $conductor,
-                            perf.primarySource = 'youtube-test'
+                            perf.primarySource = $primarySource
                         """)
                         .bind(Map.of(
                                 "id", TEST_PERFORMANCE_ID,
@@ -132,7 +132,8 @@ private void createTestData() {
                                 "videoId", TEST_VIDEO_ID,
                                 "performer", "",
                                 "ensemble", "Berlin Philharmonic",
-                                "conductor", "Herbert von Karajan"
+                                "conductor", "Herbert von Karajan",
+                                "primarySource", "youtube-test"
                         ))
         );