Data shape mismatch with nested plates

[bantin]

Thanks @fehiepsi! Adding the dims keyword arg fixed things for me.