Fixed a refactor dependant error in funtion arguments