Pass the block to the source method

Without this the find method stop of returning results if the block version is used.