Fix time setting logic for other likelihood functions