Fix incomplete test