Fix loss of shape in README.