Fix locate function

The locate function kept waiting for the device to update, but since update_devices recreates the self.devices list each time it runs, we need to refetch the device from the list.