fixed compilation error in the last commit