diff --git a/tests/test-api-importers.php b/tests/test-api-importers.php
index 8c578c06d..98b7a0a93 100644
--- a/tests/test-api-importers.php
+++ b/tests/test-api-importers.php
@@ -30,45 +30,42 @@ class TAINACAN_REST_Importers_Controller extends TAINACAN_UnitApiTestCase {
         
     }
     
-    public function test_update() {
+	public function test_update() {
+		global $Tainacan_Importer_Handler;
+		$slug='csv';
+		if ($importer = $Tainacan_Importer_Handler->initialize_importer($slug)) {
+				$response = $importer->_to_Array();
+				$Tainacan_Importer_Handler->save_importer_instance($importer);
+		} else {
+			$this->assertFalse(true);
+		}
+		$session_id = $importer->get_id();
 
-
-        $importer = new \Tainacan\Importer\CSV();
-        $session_id = $importer->get_id();
-
-        $params = json_encode([
-            'url'       => 'http://test.com',
-            'options'   => ['delimiter' => ';'],
-            'collection' => [
-                'id' => 231,
-                'map' => [
-                    30 => 'column1',
-                    31 => 'column2'
-                ],
-                'total_items' => 1234
-            ]
+		$params = json_encode([
+			'url'       => 'http://test.com',
+			'options'   => ['delimiter' => ';'],
+			'collection' => [
+					'id' => 231,
+					'map' => [
+							30 => 'column1',
+							31 => 'column2'
+					],
+					'total_items' => 1234
+			]
 		]);
-
-        $request  = new \WP_REST_Request('PATCH', $this->namespace . '/importers/session/' . $session_id);
+		$request  = new \WP_REST_Request('PATCH', $this->namespace . '/importers/session/' . $session_id);
 		$request->set_body($params);
 
-        $response = $this->server->dispatch($request);
-        $data = $response->get_data();
+		$response = $this->server->dispatch($request);
+		$data = $response->get_data();
 
-        $this->assertEquals(200, $response->get_status());
-        
-        $__importer = $_SESSION['tainacan_importer'][$session_id];
-
-        $this->assertEquals('http://test.com', $__importer->get_url());
-        $this->assertEquals(';', $__importer->get_option('delimiter'));
-
-        $this->assertEquals($__importer->get_url(), $data['url']);
-
-        $this->assertEquals(1234, $__importer->get_collections()[0]['total_items']);
-
-		
-
-    }
+		$this->assertEquals(200, $response->get_status());
+		$__importer = $Tainacan_Importer_Handler->get_importer_instance_by_session_id($session_id);
+		$this->assertEquals('http://test.com', $__importer->get_url());
+		$this->assertEquals(';', $__importer->get_option('delimiter'));
+		$this->assertEquals($__importer->get_url(), $data['url']);
+		$this->assertEquals(1234, $__importer->get_collections()[0]['total_items']);
+	}
 
 }