Minor fix for passing trainee id