Fixed missed case from the previous commit