Debugged JHJx and found another implicit reshape