TF-4645 Update logic unit test