Fix for error in previous commit