Fixed silly mistake with arguments