Removing TODO

Resolved by previous commit