Fixed issue in previous commit