diff --git a/test/Blog.Tests/Unit/Feed.spec.razor b/test/Blog.Tests/Unit/Feed.spec.razor
index 708f372..e4a4244 100644
--- a/test/Blog.Tests/Unit/Feed.spec.razor
+++ b/test/Blog.Tests/Unit/Feed.spec.razor
@@ -1,6 +1,6 @@
-@inherits Bunit.TestContext
+@inherits Bunit.BunitContext
 
-@code 
+@code
 {
   private Mock<IPostService> _postServiceMock = new();
 
@@ -8,10 +8,10 @@
   public void Feed_WhenRendedAndNoPosts_ItShouldRenderNoPostsMessage()
   {
     _postServiceMock
-      .Setup(x => x.GetPostsAsync())
-      .ReturnsAsync(new List<Post>());
+    .Setup(x => x.GetPostsAsync())
+    .ReturnsAsync(new List<Post>());
 
-    var ctx = new TestContext();
+    var ctx = new BunitContext();
 
     ctx.Services.AddSingleton(_postServiceMock.Object);
 
@@ -24,15 +24,15 @@
   public void Feed_WhenRendedAndPosts_ItShouldRenderPosts()
   {
     List<Post> posts = [
-      new Post() { Title = "Post 1", },
-      new Post() { Title = "Post 2", },
-    ];
+    new Post() { Title = "Post 1", },
+new Post() { Title = "Post 2", },
+];
 
     _postServiceMock
-      .Setup(x => x.GetPostsAsync())
-      .ReturnsAsync(posts);
+    .Setup(x => x.GetPostsAsync())
+    .ReturnsAsync(posts);
 
-    var ctx = new TestContext();
+    var ctx = new BunitContext();
 
     ctx.Services.AddScoped(_ => _postServiceMock.Object);
 
diff --git a/test/Blog.Tests/Unit/Footer.spec.razor b/test/Blog.Tests/Unit/Footer.spec.razor
index 80e5717..a774944 100644
--- a/test/Blog.Tests/Unit/Footer.spec.razor
+++ b/test/Blog.Tests/Unit/Footer.spec.razor
@@ -1,11 +1,11 @@
-@inherits Bunit.TestContext
+@inherits Bunit.BunitContext
 
-@code 
+@code
 {
   [Test]
   public void Footer_WhenRendered_ItShouldRenderFooterWithNavLinks()
   {
-    var ctx = new TestContext();
+    var ctx = new BunitContext();
 
     var cut = ctx.Render(@<Footer />);
 
diff --git a/test/Blog.Tests/Unit/Header.spec.razor b/test/Blog.Tests/Unit/Header.spec.razor
index 7963c28..d7edf73 100644
--- a/test/Blog.Tests/Unit/Header.spec.razor
+++ b/test/Blog.Tests/Unit/Header.spec.razor
@@ -1,4 +1,4 @@
-@inherits Bunit.TestContext
+@inherits Bunit.BunitContext
 
 @code
 {
@@ -29,4 +29,4 @@
     var image = cut.Find("img[alt='Stevan Freeborn']");
     image.Should().NotBeNull();
   }
-}
+}
\ No newline at end of file