Fix code/documentation mismatch