small change to rbm