add a test for this new functionality please