Corrected issue with previous commit