You could have just added the `yield` keyword (the previous version returned `Unit`)