BugFix for incremental training