diff --git a/composer.json b/composer.json
index cb02751..675169d 100644
--- a/composer.json
+++ b/composer.json
@@ -2,7 +2,6 @@
     "name": "reset-button/a-parser-php-client",
     "description": "PHP client for A-Parser API",
     "type": "library",
-    "version": "1.0.0-RC1",
     "license": "MIT",
     "autoload": {
         "psr-4": {
@@ -20,7 +19,7 @@
             "email": "counterfate@gmail.com"
         }
     ],
-    "minimum-stability": "RC",
+    "minimum-stability": "dev",
     "prefer-stable": true,
     "require": {
         "php" : "^8.1",
diff --git a/src/Actions/GetTaskStateAction.php b/src/Actions/GetTaskStateAction.php
index 2ca3636..9d2b189 100644
--- a/src/Actions/GetTaskStateAction.php
+++ b/src/Actions/GetTaskStateAction.php
@@ -6,7 +6,11 @@ class GetTaskStateAction extends Action
 {
     public const NAME = "getTaskState";
 
-    public function __construct(array $ids)
+    /**
+    * @param int|int[] $ids
+    */
+
+    public function __construct(int|array $ids)
     {
         $this->data = ['taskUid' => $ids];
     }
diff --git a/tests/Feature/Actions/GetTaskStateActionTest.php b/tests/Feature/Actions/GetTaskStateActionTest.php
index d7bd253..8620464 100644
--- a/tests/Feature/Actions/GetTaskStateActionTest.php
+++ b/tests/Feature/Actions/GetTaskStateActionTest.php
@@ -13,9 +13,17 @@ protected function setUp(): void
         parent::setUp();
     }
 
-    public function testJson()
+    public function testJsonSingleId()
     {
-        $expected = $this->prepareJson('GetTaskStateAction.json');
+        $expected = $this->prepareJson('GetTaskStateActionSingleId.json');
+
+        $action = new GetTaskStateAction(181);
+        $this->assertJsonStringEqualsJsonString($this->aparser->getJsonString($action), $expected);
+    }
+
+    public function testJsonArrayOfIds()
+    {
+        $expected = $this->prepareJson('GetTaskStateActionArrayOfIds.json');
 
         $action = new GetTaskStateAction([22,23,31]);
         $this->assertJsonStringEqualsJsonString($this->aparser->getJsonString($action), $expected);
diff --git a/tests/Fixtures/Actions/GetTaskStateAction.json b/tests/Fixtures/Actions/GetTaskStateActionArrayOfIds.json
similarity index 100%
rename from tests/Fixtures/Actions/GetTaskStateAction.json
rename to tests/Fixtures/Actions/GetTaskStateActionArrayOfIds.json
diff --git a/tests/Fixtures/Actions/GetTaskStateActionSingleId.json b/tests/Fixtures/Actions/GetTaskStateActionSingleId.json
new file mode 100644
index 0000000..ad22605
--- /dev/null
+++ b/tests/Fixtures/Actions/GetTaskStateActionSingleId.json
@@ -0,0 +1,7 @@
+{
+    "password": "{{PASSWORD}}",
+    "action": "getTaskState",
+    "data": {
+        "taskUid": 181
+    }
+}
\ No newline at end of file