fix issue with multiple state classes on the same story