Small error correction

Small error correction