Fix tests
Fix test (wrong assumptions)