Small fix to time computation