minor fix
slight fix