Fix baseline_test