fixed a bug in previous commit.. there was more states than I had thought