Fixed a bug with the context.