minor correctness change