fix to compatible with new Stream2 with .push()

Fix to support `.push()` in .`transform()`.