Create separate test for this functionality.