Fix broadcasting test