Fixes error in compute_time

See #4