Fixed assign story bug