CE-1019 Updating unit test