Fix some bug that make errors in runtime for the second layer learner.