Fix tests.
Fix sbt file.
Merge pull request #232 from AI-comp/fix_tests

Fix tests.