tidying, minor fix