fix find_disk function
