Fix controller spec generation

Introduced by #135
Merge pull request #137 from passy/test-new-fix

Fix controller spec generation