Passing other attributes to actions used by AshStateMachine

Thanks @almirsarajcic and @zachdaniel. Both solutions work!