diff --git a/generate-flatpak.py b/generate-flatpak.py
index 3a22ff8..da46c50 100644
--- a/generate-flatpak.py
+++ b/generate-flatpak.py
@@ -7,4 +7,7 @@ file.insert(0, "#!/usr/bin/env python3")
 file = "\n".join(file)
 open("sweecrypt-gtk-flatpak.py", "w").write(file)
 os.system("chmod 1777 sweecrypt-gtk-flatpak.py")
+file = open("sweecrypt-gtk.ui").read().split("\n")
+file[19] = "    <property name=\"icon-name\">codes.swee.SweeCryptGTK</property>"
+open("flatpak.ui", "w").write(file)
 os.system("mkdir flatpak")
\ No newline at end of file