Fixes the bug that caused the huge skew in training results