Fix BaggingMachine mock test