Fixing bug: predictions for --no-test-header flag failed