Fix for new rust