Fixes bug and improves comment.