Small fix on the wall drawing code