still fixing the collection not defined prolem