Temporary  fix for checkpoint warning

Temporary  fix for checkpoint warning