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