Restore accidentally removed function