Fix unit test to account for device backends