Fix Collection view controller