tiny correction in code