Added another fix that I missed