Cleanup
Fixes the specs