Fix imports in loading driver