Add test for device