Fixing dimensions bug