We're overriding the function so we need to add it in here.