Fixed forgotten hard coded function argument code