diff --git a/app/Models/Team.php b/app/Models/Team.php
index c2749d2..1396952 100644
--- a/app/Models/Team.php
+++ b/app/Models/Team.php
@@ -9,7 +9,7 @@ class Team extends Model
 {
     use HasFactory;
 
-    protected $fillable = ['name', 'size'];
+    protected $fillable = ['name', 'size', 'country_id'];
 
     public function users()
     {
diff --git a/tests/Feature/RelationshipsTest.php b/tests/Feature/RelationshipsTest.php
index 933de3b..c7bc63f 100644
--- a/tests/Feature/RelationshipsTest.php
+++ b/tests/Feature/RelationshipsTest.php
@@ -106,6 +106,7 @@ class RelationshipsTest extends TestCase
 
         $response = $this->get('/countries');
         $response->assertSee('avg team size 4');
+        $response->assertStatus(200);
     }
 
     // TASK: polymorphic relations