Fix compiler error caused by last commit