Add todo for tests