Add some test
Fixed wrong callback args