Add context class to prevent model classes being evaluated