diff --git a/tests/TestCase/Model/Behavior/VersionBehaviorTest.php b/tests/TestCase/Model/Behavior/VersionBehaviorTest.php
index 7ea5fd1..de780e7 100644
--- a/tests/TestCase/Model/Behavior/VersionBehaviorTest.php
+++ b/tests/TestCase/Model/Behavior/VersionBehaviorTest.php
@@ -468,13 +468,13 @@ public function testAssociations()
         );
         $table->addBehavior('Josegonzalez/Version.Version');
 
-        $this->assertTrue($table->associations()->has('articleversion'));
-        $versions = $table->getAssociation('articleversion');
+        $this->assertTrue($table->associations()->has('ArticleVersion'));
+        $versions = $table->getAssociation('ArticleVersion');
         $this->assertInstanceOf('Cake\Orm\Association\HasMany', $versions);
         $this->assertEquals('__version', $versions->getProperty());
 
-        $this->assertTrue($table->associations()->has('articlebodyversion'));
-        $bodyVersions = $table->getAssociation('articlebodyversion');
+        $this->assertTrue($table->associations()->has('ArticleBodyVersion'));
+        $bodyVersions = $table->getAssociation('ArticleBodyVersion');
         $this->assertInstanceOf('Cake\Orm\Association\HasMany', $bodyVersions);
         $this->assertEquals('body_version', $bodyVersions->getProperty());
     }