Fix tests for lazy DEVICE_MODEL loading