Revert "fix wrong state space of W"

This reverts commit 5a8432c5740622a67198ae0ae097938a06baec5e.