Fix little issue with new react api