Bug fix - now works with old RBM networks