Seems to be working at all. Without reshape.