Add module to mock state in tests
Add module to mock state in tests