fix some stuff related to failure model and use mock learner in test