Replaced tests with CallbackModel