Updating unit testing