small tweak to timing code