diff --git a/src/main/java/de/prob2/jupyter/commands/FindCommand.java b/src/main/java/de/prob2/jupyter/commands/FindCommand.java
index 2abd00ad89d3156933d90119e011ab103ef16849..cd5b7dbb640d96cc4ebaba5a2d7b9032197cc00c 100644
--- a/src/main/java/de/prob2/jupyter/commands/FindCommand.java
+++ b/src/main/java/de/prob2/jupyter/commands/FindCommand.java
@@ -40,8 +40,10 @@ public final class FindCommand implements Command {
 	@Override
 	public @NotNull DisplayData run(final @NotNull String argString) {
 		final Trace trace = this.animationSelector.getCurrentTrace();
-		final IEvalElement pred = trace.getModel().parseFormula(argString, FormulaExpand.EXPAND);
-		final Trace newTrace = trace.getStateSpace().getTraceToState(pred);
+		final Trace newTrace = CommandUtils.withSourceCode(argString, () -> {
+			final IEvalElement pred = trace.getModel().parseFormula(argString, FormulaExpand.EXPAND);
+			return trace.getStateSpace().getTraceToState(pred);
+		});
 		this.animationSelector.changeCurrentAnimation(newTrace);
 		return new DisplayData("Found state: " + newTrace.getCurrentState().getId());
 	}