Ok I'll do that. I will also add a test case to cover it.