Adjust tests for commit 3ec96cb038853f0f6d8198461b20c4a784cc9bf0

Signed-off-by: Madhura Jayaratne <madhura.cj@gmail.com>