Minor bugfix on dimensions