Clean up runBeforeLoop API