Fix bug (minor)