small change to make it work