small changes that make it work

Signed-off-by: volterr <volterr@gmail.com>