Update unit testing for timeline.go