diff --git a/pages/cart_page.py b/pages/cart_page.py
index 9b444b5..f7432e6 100644
--- a/pages/cart_page.py
+++ b/pages/cart_page.py
@@ -14,19 +14,25 @@ def product_box(self):
         return self.find_by_page_locator('[data-ta="product-box"]')
 
     def product_name(self):
-        return self.find_by_locator(self.product_box(), '[data-ta="product-title"]')
+        return self.find_by_locator(
+            self.product_box(), '[data-ta="product-title"]'
+        )
 
     def product_price(self):
-        return self.find_by_locator(self.product_box(), '[data-ta="product-main-price"]')
+        return self.find_by_locator(
+            self.product_box(), '[data-ta="product-main-price"]'
+        )
 
     def quantity_box(self):
         return self.find_by_page_locator('[data-ta="quantity-box"]')
 
     def quantity_box_value(self):
-        return self.find_by_locator(self.quantity_box(), 'input')
+        return self.find_by_locator(self.quantity_box(), "input")
 
     def quantity_box_remove_button(self):
-        return self.find_by_locator(self.quantity_box(), '[data-ta="quantity-remove"]')
+        return self.find_by_locator(
+            self.quantity_box(), '[data-ta="quantity-remove"]'
+        )
 
     def empty_cart_section_title(self):
-        return self.find_by_page_locator('section[class*="CartApp"]')
+        return self.find_by_page_locator('[data-ta="empty-cart-section"]')
diff --git a/pages/main_page.py b/pages/main_page.py
index 81b410a..c78c2fe 100644
--- a/pages/main_page.py
+++ b/pages/main_page.py
@@ -11,7 +11,9 @@ def go_to(self):
         self._page.goto(self._url)
 
     def popup(self):
-        return self.find_by_page_locator('[class*="CookiesConsentsBannerRodo-module_containerInfo"]')
+        return self.find_by_page_locator(
+            '[class*="CookiesConsentsBannerRodo-module_containerInfo"]'
+        )
 
     def popup_text_element(self):
         return self.find_by_text("Prywatność Użytkownika")
@@ -21,11 +23,12 @@ def desktop_buttons(self):
 
     def accept_button(self):
         return self.find_by_page_locator(
-            '[class*="InfoPage-module_buttonsDesktop"] [data-ta="cookie-btn-accept-all"]'
+            '[class*="InfoPage-module_buttonsDesktop"] '
+            '[data-ta="cookie-btn-accept-all"]'
         )
 
     def user_menu(self):
-        return self.find_by_page_locator('.empikNav__userLink.userMenu')
+        return self.find_by_page_locator(".empikNav__userLink.userMenu")
 
     def register_link(self):
         return self.find_by_page_locator('[title="Zarejestruj się"]')
@@ -34,10 +37,14 @@ def search_input(self):
         return self.find_by_page_locator('[type="search"]')
 
     def search_dropdown_related_products_list(self):
-        return self.find_by_page_locator('.css-26b661')
+        return self.find_by_page_locator('[data-ta="related-products-list"]')
 
     def search_dropdown_related_products_list_1st_value(self):
-        return self.find_by_locator(self.search_dropdown_related_products_list(), "i").first
+        return self.find_by_locator(
+            self.search_dropdown_related_products_list(), "i"
+        ).first
 
     def search_button(self):
-        return self.find_by_page_locator('//div[@class="main-search__container"]//button[@type="submit"]')
+        return self.find_by_page_locator(
+            '//div[@class="main-search__container"]//button[@type="submit"]'
+        )