Improved test, fixed callback name