fix reference to wrong collection classes