minor correction in code