fix story class