Fixed small error with collections