Fix typo
Improving test a bit