Switch to using a state like enum flag thing.