Updating: adding a minor fix