Fix for timeline tests