Fix widget height issue