Tweak timings code