Fixes compilation failure from previous commit