Fixed a small coding error with the new addition