bugfixes in arg id training