diff --git a/shared/dataflow/codeql/dataflow/internal/DataFlowImpl.qll b/shared/dataflow/codeql/dataflow/internal/DataFlowImpl.qll
index ed52b473c4b..6883b385284 100644
--- a/shared/dataflow/codeql/dataflow/internal/DataFlowImpl.qll
+++ b/shared/dataflow/codeql/dataflow/internal/DataFlowImpl.qll
@@ -169,7 +169,8 @@ module MakeImpl<LocationSig Location, InputSig<Location> Lang> {
   }
 
   /**
-   * Constructs a data flow computation given a full input configuration.
+   * Constructs a data flow computation given a full input configuration, and
+   * an initial stage 1 pruning.
    */
   module Impl<FullStateConfigSig Config, Stage1Output<Config::FlowState> Stage1> {
     private class FlowState = Config::FlowState;
diff --git a/shared/dataflow/codeql/dataflow/internal/DataFlowImplStage1.qll b/shared/dataflow/codeql/dataflow/internal/DataFlowImplStage1.qll
index ef68fcf4d22..7d786f5fa9f 100644
--- a/shared/dataflow/codeql/dataflow/internal/DataFlowImplStage1.qll
+++ b/shared/dataflow/codeql/dataflow/internal/DataFlowImplStage1.qll
@@ -460,7 +460,6 @@ module MakeImplStage1<LocationSig Location, InputSig<Location> Lang> {
 
       private class Cc = boolean;
 
-      /* Begin: Stage 1 logic. */
       /**
        * Holds if `node` is reachable from a source.
        *
@@ -954,7 +953,6 @@ module MakeImplStage1<LocationSig Location, InputSig<Location> Lang> {
             callEdgeReturn(call, c, _, _, _, _)
           )
       }
-      /* End: Stage 1 logic. */
     }
 
     private module Stage1Common {