Fixed Signal Issue in the trainning