Fixes selection of n_shape for LKAAMFitters