Fix compilation since previous commit