Fixed failing tests

Signed-off-by: Chanaka Indrajith <pe.chanaka.ck@gmail.com>