Update other testing code

Signed-off-by: Jacky Boen <aqiank@gmail.com>