adding the fix by only returning what is desired in the data function