add two tests, fix index