Adding the test, unfortunately it is not working correctly.