fix for loading states