made it work for the example