diff --git a/python/lsst/ap/association/__init__.py b/python/lsst/ap/association/__init__.py
index f91a0347..66e990b8 100644
--- a/python/lsst/ap/association/__init__.py
+++ b/python/lsst/ap/association/__init__.py
@@ -27,3 +27,4 @@
 from .packageAlerts import *
 from .diaPipe import *
 from .transformDiaSourceCatalog import *
+from .filterDiaSourceCatalog import *
diff --git a/python/lsst/ap/association/filterDiaSourceCatalog.py b/python/lsst/ap/association/filterDiaSourceCatalog.py
index 7a5fa325..9d3832d9 100644
--- a/python/lsst/ap/association/filterDiaSourceCatalog.py
+++ b/python/lsst/ap/association/filterDiaSourceCatalog.py
@@ -85,7 +85,7 @@ def runQuantum(self, butlerQC, inputRefs, outputRefs):
         butlerQC.put(outputs, outputRefs)
 
     @timeMethod
-    def run(self, diaSourceCat):
+    def run(self, diaSourceCat, ccdVisitId):
         """Filter sky sources from the supplied DiaSource catalog.
 
         Parameters
diff --git a/python/lsst/ap/association/transformDiaSourceCatalog.py b/python/lsst/ap/association/transformDiaSourceCatalog.py
index f393ec90..566ed8c2 100644
--- a/python/lsst/ap/association/transformDiaSourceCatalog.py
+++ b/python/lsst/ap/association/transformDiaSourceCatalog.py
@@ -49,7 +49,7 @@ class TransformDiaSourceCatalogConnections(pipeBase.PipelineTaskConnections,
     )
     diaSourceCat = connTypes.Input(
         doc="Catalog of DiaSources produced during image differencing.",
-        name="{fakesType}{coaddName}Diff_diaSrc",
+        name="{fakesType}{coaddName}Diff_candidateDiaSrc",
         storageClass="SourceCatalog",
         dimensions=("instrument", "visit", "detector"),
     )