small fix for TODO