diff --git a/src/components/badge/BadgeDirective.js b/src/components/badge/BadgeDirective.js
index 279bf2547..804900dcb 100644
--- a/src/components/badge/BadgeDirective.js
+++ b/src/components/badge/BadgeDirective.js
@@ -34,7 +34,7 @@ const BadgeDirective = {
         if (options.oldValue !== options.value) {
             let badge = document.getElementById(el.$_pbadgeId);
 
-            if (options.value && DomHandler.hasClass('p-badge-dot')) {
+            if (options.value && DomHandler.hasClass(badge, 'p-badge-dot')) {
                 DomHandler.removeClass(badge, 'p-badge-dot');
 
                 if (String(options.value).length === 1)
@@ -42,7 +42,7 @@ const BadgeDirective = {
                 else
                     DomHandler.removeClass(badge, 'p-badge-no-gutter');
             }
-            else if (!options.value && !DomHandler.hasClass('p-badge-dot')) {
+            else if (!options.value && !DomHandler.hasClass(badge, 'p-badge-dot')) {
                 DomHandler.addClass(badge, 'p-badge-dot');
             }