fix last test, TODO in test