Fix for build, broken by the previous commit