Fixed tests after loading states