Fix rust tests