diff --git a/feedback-ui-design/script.js b/feedback-ui-design/script.js
index b3eb3fd..1b758c5 100644
--- a/feedback-ui-design/script.js
+++ b/feedback-ui-design/script.js
@@ -5,15 +5,18 @@ const panel = document.querySelector('#panel')
 let selectedRating = 'Satisfied'
 
 ratingsContainer.addEventListener('click', (e) => {
-    if(e.target.parentNode.classList.contains('rating')) {
+    if(e.target.parentNode.classList.contains('rating') && e.target.nextElementSibling) {
         removeActive()
         e.target.parentNode.classList.add('active')
         selectedRating = e.target.nextElementSibling.innerHTML
-    }
-    if(e.target.classList.contains('rating')) {
+    } else if(
+        e.target.parentNode.classList.contains('rating') &&
+        e.target.previousSibling &&
+        e.target.previousElementSibling.nodeName === 'IMG'
+    ) {
         removeActive()
-        e.target.classList.add('active')
-        selectedRating = e.target.nextElementSibling.innerHTML
+        e.target.parentNode.classList.add('active')
+        selectedRating = e.target.innerHTML
     }
 
 })
@@ -32,4 +35,4 @@ function removeActive() {
     for(let i = 0; i < ratings.length; i++) {
         ratings[i].classList.remove('active')
     }
-}
\ No newline at end of file
+}