Fixed dimension error on finding pts_W