Fixes a function that was removed