Pass callback to env render instead of the other way 'round'