Remove TODO
Update reference api