diff --git a/src/services/SitemapService.php b/src/services/SitemapService.php
index a9193cd..2397494 100644
--- a/src/services/SitemapService.php
+++ b/src/services/SitemapService.php
@@ -556,12 +556,15 @@ private function _getPageCount (Element $type, $id)
 	private function _setCriteriaIdByType ($criteria, Element $type, $id)
 	{
 		switch ($type::className()) {
+			case 'craft\\elements\\Entry':
 			case 'Entry':
 				$criteria->sectionId = $id;
 				break;
+			case 'craft\\elements\\Category':
 			case 'Category':
 				$criteria->groupId = $id;
 				break;
+			case 'craft\\elements\\Product':
 			case 'Product':
 				$criteria->typeId = $id;
 				break;