diff --git a/app/src/main/java/com/example/hpostesting/presentation/deviceprovision/DeviceProvisionViewModel.kt b/app/src/main/java/com/example/hpostesting/presentation/deviceprovision/DeviceProvisionViewModel.kt
index f2cd209..466e0d5 100644
--- a/app/src/main/java/com/example/hpostesting/presentation/deviceprovision/DeviceProvisionViewModel.kt
+++ b/app/src/main/java/com/example/hpostesting/presentation/deviceprovision/DeviceProvisionViewModel.kt
@@ -1,16 +1,12 @@
 package com.example.hpostesting.presentation.deviceprovision
 
-import android.content.Context
-import androidx.lifecycle.LiveData
 import androidx.lifecycle.MutableLiveData
 import androidx.lifecycle.ViewModel
 import androidx.lifecycle.viewModelScope
-import com.example.hpostesting.data.NetworkStatusLiveData
 import com.example.hpostesting.data.Result
 import com.example.hpostesting.data.model.Response
 import com.example.hpostesting.data.model.deviceprovision.DeviceProvisionRequest
 import com.example.hpostesting.data.model.deviceprovision.DeviceProvisionResponse
-import com.example.hpostesting.data.model.diagnostics.DiagnosticsData
 import com.example.hpostesting.data.model.patient.DeviceData
 import com.example.hpostesting.data.repository.Repository
 import dagger.hilt.android.lifecycle.HiltViewModel
@@ -19,37 +15,34 @@ import javax.inject.Inject
 
 @HiltViewModel
 class DeviceProvisionViewModel @Inject constructor(
-    private val databaseRepository: Repository,
     private val repository: Repository,
-    context: Context,
 ) : ViewModel() {
-
-    private val _networkStatusLiveData = NetworkStatusLiveData(context)
     var isServiceConnected = false
-    val networkStatusLiveData: LiveData<Boolean>   get() = _networkStatusLiveData
     val deviceProvisionResponse = MutableLiveData<Result<DeviceProvisionResponse>>()
     val fireBaseUpload = MutableLiveData<String>()
     val fireBaseBulkUpload = MutableLiveData<String>()
     val deviceData = MutableLiveData<DeviceData?>()
+
     fun deviceProvision(deviceProvisionRequest: DeviceProvisionRequest) = viewModelScope.launch {
         deviceProvisionResponse.postValue(Result.Loading())
-        databaseRepository.deviceProvision(deviceProvisionRequest).let {
+        repository.deviceProvision(deviceProvisionRequest).let {
             deviceProvisionResponse.postValue(it)
         }
     }
 
 
-    fun addDeviceId(data: DeviceData){
+    fun addDeviceId(data: DeviceData) {
         viewModelScope.launch {
-            try{
+            try {
                 repository.uploadDeviceId(data)
                 fireBaseUpload.postValue("Successfully device Id uploaded to firebase")
             } catch (e: Exception) {
 //                Log.e("Testdb", "Exception during data upload: ${e.message}")
-            fireBaseUpload.postValue("Error")
-        }
+                fireBaseUpload.postValue("Error")
+            }
         }
     }
+
     fun addDeviceProvisionDataToDb(data: DeviceData) {
         viewModelScope.launch {
             try {