Fix replace function