Fix handling arguments in after loading callbacks