diff --git a/tasks/install_ubuntu.yml b/tasks/install_ubuntu.yml
index c73fff0..4c8238c 100644
--- a/tasks/install_ubuntu.yml
+++ b/tasks/install_ubuntu.yml
@@ -5,12 +5,22 @@
     keyring: /etc/apt/trusted.gpg.d/jellyfin.gpg
 #curl -fsSL https://repo.jellyfin.org/ubuntu/jellyfin_team.gpg.key | sudo gpg --dearmor -o /etc/apt/trusted.gpg.d/jellyfin.gpg
 
+- name: Set arch
+  set_fact:
+    arch: amd64
+  when: ansible_facts['architecture'] == 'x86_64'
+
+- name: Set arch
+  set_fact:
+    arch: arm64
+  when: ansible_facts['architecture'] == 'aarch64'
+
 - name: Ensure repository is configured
   apt_repository:
-    repo: "deb [arch={{ ansible_facts['facter_os']['architecture'] }}] https://repo.jellyfin.org/{{ ansible_facts['distribution'] | lower }} {{ ansible_facts['distribution_release'] }} main"
+    repo: "deb [arch={{ ansible_facts['architecture'] }}] https://repo.jellyfin.org/{{ ansible_facts['distribution'] | lower }} {{ ansible_facts['distribution_release'] }} main"
     state: present
 #echo "deb [arch=$( dpkg --print-architecture )] https://repo.jellyfin.org/$( awk -F'=' '/^ID=/{ print $NF }' /etc/os-release ) $( awk -F'=' '/^VERSION_CODENAME=/{ print $NF }' /etc/os-release ) main" | sudo tee /etc/apt/sources.list.d/jellyfin.list
 
 - name: Ensure Jellyfin package is installed
   package:
-    name: jellyfin
\ No newline at end of file
+    name: jellyfin