Update import for check