fix unfold to be w,b