could you add a test for this?