Add tests for shallow