little fix with broadcasting