Fix for #96
Fix typo in tests