Small fix for timing