Fixed args bug in callback