Fix 'not supported on device' DeviceErrors in PennyLane | Vectle