Using the correct # of states