Just needed to move a part of a function to another to match the specs.