Issue #15: Pass the context to the onUpgrade helper