diff --git a/src/DataSet/DataSet.php b/src/DataSet/DataSet.php
index 2a605f1..1a77593 100644
--- a/src/DataSet/DataSet.php
+++ b/src/DataSet/DataSet.php
@@ -40,7 +40,7 @@ public function __construct(Connector $connection) {
 	 * see $this->fixture
 	 *
 	 * @param array $data
-	 * @return $this
+	 * @return \Zumba\PHPUnit\Extensions\Mongo\DataSet\DataSet
 	 */
 	public function setFixture(array $data) {
 		$this->fixture = $data;
@@ -50,7 +50,7 @@ public function setFixture(array $data) {
 	/**
 	 * Drops all collections specified in the fixture keys.
 	 *
-	 * @return $this
+	 * @return \Zumba\PHPUnit\Extensions\Mongo\DataSet\DataSet
 	 */
 	public function dropAllCollections() {
 		foreach ($this->fixture as $collectionKey => $collectionData) {
@@ -62,7 +62,7 @@ public function dropAllCollections() {
 	/**
 	 * Creates all collections with data from the fixture.
 	 *
-	 * @return $this
+	 * @return \Zumba\PHPUnit\Extensions\Mongo\DataSet\DataSet
 	 */
 	public function buildCollections() {
 		foreach ($this->fixture as $collection => $data) {