fix widget integration