Replace function that was removed