small fix
rewrite