diff --git a/taint_mini/taintmini.py b/taint_mini/taintmini.py index 27a60e1..a2a47c1 100644 --- a/taint_mini/taintmini.py +++ b/taint_mini/taintmini.py @@ -28,7 +28,7 @@ def filter_results(results, config): filtered[page].append(flow) else: # no sink filter, just apply source filter - if flow['sink'] in config["sinks"]: + if flow['source'] in config["sources"]: filtered[page].append(flow) else: # no source filter, apply sink filter