Add test for finding unique key