Good catch, might need a refactor of the function to allow this to be genericized but I can look into that.