Fixing the broken test by returning null if no records...will be adding a tests specifically for that