Add one more test
Remove unused if