fixing more bugs related to dimension dropping of [