More tests
Fix the test