Fixing bug with callback method.