update tests

Fix test cases