Fixed missing collection code