Fixed key error in test_snapshots.py