BUG: fix array shape
BUG: fix array shape

mnicholl/MOSFiT,guillochon/MOSFiT,mnicholl/MOSFiT,mnicholl/MOSFiT,guillochon/FriendlyFit,villrv/MOSFiT,bmockler/MOSFiT,guillochon/MOSFiT,bmockler/MOSFiT,guillochon/MOSFiT,villrv/MOSFiT