Simple fix for loading args. I don't know how this was working before