Small update to previous pull request