diff --git a/deeplc/feat_extractor.py b/deeplc/feat_extractor.py
index 509964b..6c9eb4f 100644
--- a/deeplc/feat_extractor.py
+++ b/deeplc/feat_extractor.py
@@ -463,7 +463,9 @@ def rolling_sum(a, n=2):
                     try:
                         matrix[i, dict_index[k]] = v
                     except IndexError:
-                        logger.warning(f"Could not add the following value: {v}")
+                        logger.warning(f"Could not add the following value: pos {i} for atom {k} with value {v}")
+                    except KeyError:
+                        logger.warning(f"Could not add the following value: pos {i} for atom {k} with value {v}")
 
             for p in positions_pos:
                 aa = seq[p]