Minor test improvements.

Signed-off-by: Rocky Jaiswal <rocky.jaiswal@gmail.com>