Fix compiler error from previous commit