I should clean up the TODO tho.