Fixed reshape bug