1 files changed, 12 insertions, 5 deletions
diff --git a/test/segments/ram.spec b/test/segments/ram.spec
index 0deaacf9..dc0b6c6b 100755
--- a/test/segments/ram.spec
+++ b/test/segments/ram.spec
@@ -7,8 +7,6 @@ SHUNIT_PARENT=$0
 
 function setUp() {
   export TERM="xterm-256color"
-  # Load Powerlevel9k
-  source powerlevel9k.zsh-theme
 
   P9K_HOME=$(pwd)
   ### Test specific
@@ -28,31 +26,40 @@ function tearDown() {
 }
 
 function testRamSegmentWorksOnOsx() {
-    local OS="OSX" # Fake OSX
     alias vm_stat="echo 'Mach Virtual Memory Statistics: (page size of 4096 bytes)
 Pages free:                              299687.
 Pages active:                           1623792.
 Pages inactive:                         1313411.
 '"
 
+    # Load Powerlevel9k
+    source ${P9K_HOME}/powerlevel9k.zsh-theme
+    local OS="OSX" # Fake OSX
+
     assertEquals "%K{yellow} %F{black%}RAM %f%F{black}6.15G " "$(prompt_ram left 1 false ${FOLDER})"
 
     unalias vm_stat
 }
 
 function testRamSegmentWorksOnBsd() {
-    local OS="BSD" # Fake BSD
     mkdir -p var/run
     echo "avail memory 5678B 299687 4444G 299" > var/run/dmesg.boot
 
+    # Load Powerlevel9k
+    source ${P9K_HOME}/powerlevel9k.zsh-theme
+    local OS="BSD" # Fake BSD
+
     assertEquals "%K{yellow} %F{black%}RAM %f%F{black}0.29M " "$(prompt_ram left 1 false ${FOLDER})"
 }
 
 function testRamSegmentWorksOnLinux() {
-    local OS="Linux" # Fake Linux
     mkdir proc
     echo "MemAvailable: 299687" > proc/meminfo
 
+    # Load Powerlevel9k
+    source ${P9K_HOME}/powerlevel9k.zsh-theme
+    local OS="Linux" # Fake Linux
+
     assertEquals "%K{yellow} %F{black%}RAM %f%F{black}0.29G " "$(prompt_ram left 1 false ${FOLDER})"
 }