fixed usage of reshape