Fix for app not catching early end of probation