minor fix
a bit more