diff --git a/tools/libs/set_gpumem.sh b/tools/libs/set_gpumem.sh
index e732db1..ab183e0 100755
--- a/tools/libs/set_gpumem.sh
+++ b/tools/libs/set_gpumem.sh
@@ -29,11 +29,9 @@ set_gpu_mem() {
     ## Determine Ramsize and export MAKEFLAG
     if [[ "$(get_avail_mem)" -le 524288 ]]; then
         mem_split=128
-    fi
-    if [[ "$(get_avail_mem)" -le 1048576 ]]; then
+    elif [[ "$(get_avail_mem)" -le 1048576 ]]; then
         mem_split=160
-    fi
-    if [[ "$(get_avail_mem)" -gt 1048576 ]]; then
+    else
         mem_split=256
     fi
     if [[ "$(is_raspbian)" = "1" ]] && [[ -n "$(command -v raspi-config)" ]]; then