Small update to inference section