diff --git a/test/inst_complex_welcome_test.rb b/test/inst_complex_welcome_test.rb
index 6e645269a..1b83636ff 100755
--- a/test/inst_complex_welcome_test.rb
+++ b/test/inst_complex_welcome_test.rb
@@ -22,6 +22,7 @@
     # null object returned above as parameter and it raise exception from
     # component system
     stub_const("Yast::UI", double.as_null_object)
+    allow(Yast::Pkg).to receive(:SourceGetCurrent).and_return([])
   end
 
   after do
diff --git a/test/inst_finish_test.rb b/test/inst_finish_test.rb
index ff2bc9aa1..e24056741 100755
--- a/test/inst_finish_test.rb
+++ b/test/inst_finish_test.rb
@@ -26,6 +26,7 @@
 
       allow(Yast::Pkg).to receive(:TargetInitialize)
       allow(Yast::Pkg).to receive(:TargetLoad)
+      allow(Yast::Pkg).to receive(:PkgInstalled).and_return(false)
 
       allow(Yast::Hooks).to receive(:run)
     end
diff --git a/test/update_repository_test.rb b/test/update_repository_test.rb
index 7790f9764..9f813fabb 100755
--- a/test/update_repository_test.rb
+++ b/test/update_repository_test.rb
@@ -243,8 +243,11 @@
   end
 
   describe "#cleanup" do
-    it "deletes the repository" do
+    it "deletes the repository and releases it" do
       expect(Yast::Pkg).to receive(:SourceDelete).with(repo_id)
+      expect(Yast::Pkg).to receive(:SourceReleaseAll)
+      expect(Yast::Pkg).to receive(:SourceSaveAll)
+
       subject.cleanup
     end
   end