minor fix
Merge pull request #6 from jinchung/refactoring

minor fix