[PATCH 2/2] Honor modules' __all__ when doing collect