temporary fix for belief state size
merged