Does the `StatesController` need this too?