From cfc2adcb787f405ab6d260b063c2e8ce9262b12b Mon Sep 17 00:00:00 2001
From: rftrdev <102184004+rftrdev@users.noreply.github.com>
Date: Wed, 23 Aug 2023 00:34:31 -0700
Subject: [PATCH] Fix personality check for backgrounds during IMP creation

---
 Laptop/IMP Background.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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
+}