Fixed issue when top level key doesn't exist in context