Fix a bug by @f98ae05e9 #9