diff --git a/Sources/Classes/Search/BaseSearchViewController.swift b/Sources/Classes/Search/BaseSearchViewController.swift
index c3d1fc87..feed4ebe 100644
--- a/Sources/Classes/Search/BaseSearchViewController.swift
+++ b/Sources/Classes/Search/BaseSearchViewController.swift
@@ -58,8 +58,8 @@ where ViewModel: BaseSearchViewModel<Item, ItemViewModel> {
     open override func bindViews() {
         super.bindViews()
         viewModel.itemsViewModelsDriver
-            .drive(onNext: { [weak self] viewModels in
-                self?.handle(itemViewModels: viewModels)
+            .drive(with: self, onNext: { object, viewModels in
+                object.handle(itemViewModels: viewModels)
             })
             .disposed(by: disposeBag)