small fix to mock timing