Changed code
minor fix