minor fix
almost done