Fixed an issue where features were trained too early.