Fix borrow check warning