update method and remove failed unit test

Signed-off-by: ahmadshah <ahmadshahhafizan@gmail.com>