Updating minor fix