Add ability to force a state for testing