Update reference line remove callback

The remove line callback was using an outdated method prototype.