small change
make it work