Revert "improving check_write_disk.py"

This reverts commit 1f0cd7da2d4edbfc525b017a186b27871656b7e4.