Draft: Small fix in dimensions