forgot to handle 1D case :P
1.0.1