Fix mistakes with dimensions