Fixed reshaping errors