Clean up potential stake code