diff --git a/pytiled_parser/xml_parser.py b/pytiled_parser/xml_parser.py
index efc0c4b..f770ea0 100644
--- a/pytiled_parser/xml_parser.py
+++ b/pytiled_parser/xml_parser.py
@@ -193,7 +193,7 @@ def _parse_layer(
     Returns:
         FIXME
     """
-    if "id" in layer_element:
+    if "id" in layer_element.attrib:
         id_ = int(layer_element.attrib["id"])
     else:
         id_ = None
diff --git a/tests/test_parser.py b/tests/test_parser.py
index d488e84..cb859b5 100644
--- a/tests/test_parser.py
+++ b/tests/test_parser.py
@@ -1,17 +1,13 @@
 """Unit tests for pytiled_parser"""
 
 import xml.etree.ElementTree as etree
-from contextlib import contextmanager
+from contextlib import ExitStack as does_not_raise
 
 import pytest
+
 from pytiled_parser import objects, utilities, xml_parser
 
 
-@contextmanager
-def does_not_raise():
-    yield
-
-
 def _get_root_element(xml: str) -> etree.Element:
     """Get root element of string of XML.