temp fix to have correct order in the attribution generation based on the input.