Add one more test to check the result