[~] Small fix for last update