BUG: fix test
BUG: fix test

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