fix broadcasting