Now able to mock state