fix call to context in plot