Final final commit

Minor change to one function