Fixed error in previous commit