Fix visualization of 1d model