Coding fix
Small fix