Followup fix for the lightning plugin