fix Wei789/myth#3

take a look @Wei789
fix Wei789/myth#3