add missing collection import