fix bug with collection view