Update the test

Signed-off-by: Ryan Chen <ryan.chen@turn.com>