Review tests and fix issue in the example exposed in the README