Fix a bug

Here is the PR description.