fix bug
Merge pull request #18 from OmniaGM/sw

fix bug