fix minor bug
2 files changed