Fix disk visualization