Change callback to done to match common format