removed dependency on state_transformer