diff --git a/wireguard-install.sh b/wireguard-install.sh
index d450d99..fcf1671 100644
--- a/wireguard-install.sh
+++ b/wireguard-install.sh
@@ -121,15 +121,15 @@ function installWireGuard() {
 
 	# Install WireGuard tools and module
 	if [[ ${OS} == 'ubuntu' ]]; then
-		apt-get update
-		apt-get install -y wireguard iptables resolvconf qrencode
+		apt-get update && apt-get -y upgrade
+		apt-get install -y linux-headers-$(uname -r) wireguard iptables resolvconf qrencode
 	elif [[ ${OS} == 'debian' ]]; then
 		if ! grep -rqs "^deb .* buster-backports" /etc/apt/; then
 			echo "deb http://deb.debian.org/debian buster-backports main" >/etc/apt/sources.list.d/backports.list
 			apt-get update
 		fi
-		apt update
-		apt-get install -y iptables resolvconf qrencode
+		apt update && apt-get -y upgrade
+		apt-get install -y iptables resolvconf qrencode linux-headers-$(uname -r)
 		apt-get install -y -t buster-backports wireguard
 	elif [[ ${OS} == 'fedora' ]]; then
 		if [[ ${VERSION_ID} -lt 32 ]]; then