1) Fix `yield_modules` to actually yield all modules (#402) (dycw)
(SHA: 50b5a81102b6cf07f02de15daebff7d567dd9905)