minor fix
Some minor fixes