Resolve some bugs
Merge pull request #73 from Jero786/MYG000-214

Resolve some bugs