minor fix in code arrangment