Quick edits

Fixing restrictions
Fixing a test that was dumb