Updated unit test to match code

Updated unit test to match code change in commit 4e64f9f28dfa2fd4e477e4b6c005312fae824050.