diff --git a/gui.js b/gui.js
index 7aed2166be..664a527a21 100644
--- a/gui.js
+++ b/gui.js
@@ -1920,6 +1920,12 @@ async function updated_page(no_graph_restart, disable_auto_enable_valid_layer_ty
 	last_updated_page = Date.now();
 
 	disable_everything_in_last_layer_enable_everyone_else_in_beginner_mode();
+
+	if($("#loss").val() == "categoricalCrossentropy" && is_classification) {
+		$("#download_with_data").show();
+	} else {
+		$("#download_with_data").hide();
+	}
 }
 
 async function change_optimizer() {
diff --git a/tabs/code.php b/tabs/code.php
index 426e85560d..2925b484bb 100644
--- a/tabs/code.php
+++ b/tabs/code.php
@@ -27,7 +27,7 @@
 			<button onclick="save_model()">
 				<span class="TRANSLATEME_download_model_data"></span>
 			</button>
-			<button onclick="download_model_for_training(model)">
+			<button id="download_with_data" onclick="download_model_for_training()">
 				<span class="TRANSLATEME_download_for_local_taurus"></span>
 			</button>
 		</span>