Tiny bug in dimension when using specific parameters

[Ting-Bao]

When using parameters:

max_elms = 3
min_elms = 3
max_sites = 3

the input and output dimension are different. Of course, It's only a corner case.