TODO: should be assertEquals