diff --git a/farm_test.go b/farm_test.go
index e443d65..1445f45 100644
--- a/farm_test.go
+++ b/farm_test.go
@@ -195,4 +195,11 @@ func TestFarmMethods(t *testing.T) {
 			t.Errorf("Expected 2 llamas, got %d", len(llamas))
 		}
 	})
+
+	t.Run("TestOllamaFarm", func(t *testing.T) {
+		llama := farm.First(&ollamafarm.Where{Group: "group1"})
+		if llama.Farm() != farm {
+			t.Error("Expected farm, got nil")
+		}
+	})
 }