Fixed a missing argument bug i introduced