Updated to match already existing code.