Make argument and parameter registers work properly