done
minor fix