fixing parallel training