diff --git a/aws-organizations-organizationalunit/src/test/java/software/amazon/organizations/organizationalunit/UpdateHandlerTest.java b/aws-organizations-organizationalunit/src/test/java/software/amazon/organizations/organizationalunit/UpdateHandlerTest.java
index 5f80c32..1197bb5 100644
--- a/aws-organizations-organizationalunit/src/test/java/software/amazon/organizations/organizationalunit/UpdateHandlerTest.java
+++ b/aws-organizations-organizationalunit/src/test/java/software/amazon/organizations/organizationalunit/UpdateHandlerTest.java
@@ -1,9 +1,7 @@
 package software.amazon.organizations.organizationalunit;
 
 import java.time.Duration;
-import java.util.Collection;
 import java.util.HashSet;
-import java.util.List;
 import java.util.Set;
 
 import software.amazon.awssdk.services.organizations.OrganizationsClient;
@@ -89,12 +87,12 @@ public void handleRequest_SimpleSuccess() {
 
         final Set<Tag> oldTags = TagsHelper.mergeTags(
                 TagsHelper.convertOrganizationalUnitTagToOrganizationTag(previousResourceModel.getTags()),
-                TagTestResourcesHelper.defaultStackTags
+                request.getPreviousResourceTags()
         );
 
         final Set<Tag> newTags = TagsHelper.mergeTags(
                 TagsHelper.convertOrganizationalUnitTagToOrganizationTag(model.getTags()),
-                TagTestResourcesHelper.updatedStackTags
+                request.getDesiredResourceTags()
         );
 
         final Set<Tag> tagsToAddOrUpdate = TagsHelper.getTagsToAddOrUpdate(oldTags, newTags);