I'll add the test to this code, so we could ensure that this is working as expected!