Small fix to test for distinct