Fixes the after penalty bug.