Fix figure building for timebag evolution