fix: : could not broadcast input array from shape (2,) into shape (1,)

fix: : could not broadcast input array from shape (2,) into shape (1,) #414