diff --git a/kiwi/builder/install.py b/kiwi/builder/install.py
index 8cb8cf2e..592be974 100644
--- a/kiwi/builder/install.py
+++ b/kiwi/builder/install.py
@@ -280,7 +280,7 @@ class InstallImageBuilder(object):
                 self.xml_state.xml_data.get_name(), '.md5'
             ]
         )
-        checksum = Checksum(pxe_image_filename)
+        checksum = Checksum(self.diskname)
         checksum.md5(pxe_md5_filename)
 
         # the kiwi initrd code triggers the install by trigger files
diff --git a/test/unit/builder_install_test.py b/test/unit/builder_install_test.py
index e31739ee..e9fc66c5 100644
--- a/test/unit/builder_install_test.py
+++ b/test/unit/builder_install_test.py
@@ -255,7 +255,7 @@ class TestInstallImageBuilder(object):
             ['mv', compress.compressed_filename, 'tmpdir/result-image.xz']
         )
         mock_md5.assert_called_once_with(
-            'tmpdir/result-image.xz'
+            'target_dir/result-image.x86_64-1.2.3.raw'
         )
         checksum.md5.assert_called_once_with(
             'tmpdir/result-image.md5'