Fix wrong import in collection