Fixed wrong assertions in tests