From 9b9cf7ca527aa9d73386d306a1a25a315e0610d7 Mon Sep 17 00:00:00 2001
From: Théo de la Hogue
Date: Tue, 5 Sep 2023 08:48:44 +0200
Subject: Drawing background only if heatmap object is None too.

---
 src/argaze/ArFeatures.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/argaze/ArFeatures.py b/src/argaze/ArFeatures.py
index f40377b..1a13092 100644
--- a/src/argaze/ArFeatures.py
+++ b/src/argaze/ArFeatures.py
@@ -1019,7 +1019,7 @@ class ArFrame():
 		self.__look_lock.acquire()
 
 		# Draw background only
-		if background_weight is not None and heatmap_weight is None:
+		if background_weight is not None and (heatmap_weight is None or self.heatmap is None):
 
 			image = self.background.copy()
 
@@ -1069,7 +1069,6 @@ class ArFrame():
 		Parameters:
 			kwargs: ArFrame.__image parameters
 		"""
-
 		# Use image_parameters attribute if no kwargs
 		if kwargs:
 
-- 
cgit v1.1