Use last correct test to calculate total_model_time.

Fixes #12

Reviewers: @mhyee