Fix failing boltzmann_machine tests