fix tests - checkpoint
fixed test error and added test - checkpoint