diff --git a/save_load/ui/save_file.gd b/save_load/ui/save_file.gd
index 976b6f8..35e0aec 100644
--- a/save_load/ui/save_file.gd
+++ b/save_load/ui/save_file.gd
@@ -38,8 +38,9 @@ func set_save_icon() -> void:
 	save_icon.texture = load(save_file_details.save_icon)
 
 func set_save_name() -> void:
-	# TODO: Remove extension from name
-	save_name_label.text = save_file_details.filename
+	save_name_label.text = save_file_details.save_name
+	if save_file_details.save_name == null or save_file_details.save_name == "":
+		save_name_label.text = save_file_details.filename