Small fixes to lost counter implementation