fix unit testing todos