Fix passing bad context for callback function