Added State management test