Added a couple of contract tests