Align to the most recent go-statemachine

Signed-off-by: Ondrej Kupka <ondra.cap@gmail.com>