fix: removing unnecesary dimensions