Fix imports to make them work properly