diff --git a/Laptop/IMP Background.cpp b/Laptop/IMP Background.cpp
index 9ea5c40e7..b7cddab42 100644
--- a/Laptop/IMP Background.cpp	
+++ b/Laptop/IMP Background.cpp	
@@ -15,6 +15,7 @@
 	#include "CharProfile.h"
 	#include "soldier profile type.h"
 	#include "IMP Compile Character.h"
+	#include "IMP Disability Trait.h"
 	#include "GameSettings.h"
 	#include "Interface.h"
 
@@ -671,7 +672,7 @@ BOOLEAN IsBackGroundAllowed( UINT16 ubNumber )
 			return FALSE;
 	}
 
-	switch ( iPersonality )
+	switch ( iChosenDisabilityTrait() )
 	{
 	case HEAT_INTOLERANT:
 		if ( zBackground[ ubNumber ].value[BG_DESERT] > 0 )
@@ -807,4 +808,4 @@ void BtnIMPBackgroundPreviousCallback(GUI_BUTTON *btn,INT32 reason)
 			usBackground = 0;
 		}
 	}
-}
\ No newline at end of file
+}