diff --git a/smc/src/main/java/com/tlcsdm/smc/tools/FileDiff.java b/smc/src/main/java/com/tlcsdm/smc/tools/FileDiff.java
index 723e27f7c..ad7c96dd0 100644
--- a/smc/src/main/java/com/tlcsdm/smc/tools/FileDiff.java
+++ b/smc/src/main/java/com/tlcsdm/smc/tools/FileDiff.java
@@ -51,6 +51,7 @@ public class FileDiff extends SmcSample {
 		webView.getEngine().loadContent(template);
 		notificationBuilder.text(I18nUtils.get("smc.tool.fileDiff.button.generate.success"));
 		notificationBuilder.showInformation();
+		bindUserData();
 	});
 
 	private final Action download = new Action(I18nUtils.get("smc.tool.button.download"), actionEvent -> {
@@ -64,6 +65,7 @@ public class FileDiff extends SmcSample {
 		DiffHandleUtils.generateDiffHtml(diffString, outputField.getText() + "\\diff.html");
 		notificationBuilder.text(I18nUtils.get("smc.tool.fileDiff.button.download.success"));
 		notificationBuilder.showInformation();
+		bindUserData();
 	});
 
 	private final Collection<? extends Action> actions = List.of(generate, download);
@@ -144,6 +146,10 @@ public Node getPanel(Stage stage) {
 		webView.setMaxSize(Double.MAX_VALUE, Double.MAX_VALUE);
 		GridPane.setVgrow(webView, Priority.ALWAYS);
 
+		userData.put("original", originalField);
+		userData.put("compare", compareField);
+		userData.put("output", outputField);
+
 		grid.add(toolBar, 0, 0, 3, 1);
 		grid.add(originalLabel, 0, 1);
 		grid.add(originalButton, 1, 1);