Fix broadcasting error after outer product change