trying UI fix for states