Clean up the context mess.

Signed-off-by: katlogic <kat@lua.cz>