X-Git-Url: https://git.mxchange.org/?a=blobdiff_plain;f=application%2Fhub%2Fmain%2Fstates%2Fdht%2Fclass_BaseDhtState.php;fp=application%2Fhub%2Fmain%2Fstates%2Fdht%2Fclass_BaseDhtState.php;h=78b10f3cf47b4571918a464ee4b3a0af657c0c08;hb=35871df2ede1f1dc12223601e268ef37b4cfc0f6;hp=88c2de0c645ca90128b5fbb1ecd21ac0e8b47fc0;hpb=b3ccc82c835c0aaa1de13a34f1ed222ee82f2fb6;p=hub.git

diff --git a/application/hub/main/states/dht/class_BaseDhtState.php b/application/hub/main/states/dht/class_BaseDhtState.php
index 88c2de0c6..78b10f3cf 100644
--- a/application/hub/main/states/dht/class_BaseDhtState.php
+++ b/application/hub/main/states/dht/class_BaseDhtState.php
@@ -41,9 +41,9 @@ class BaseDhtState extends BaseState {
 	 * @return	void
 	 * @throws	InvalidStateException	If the state is not 'virgin'
 	 */
-	public function validateDhtStateIsVirgin () {
+	public function validateDhtStateIsVirginOrInit () {
 		// Just compare it...
-		if (!$this instanceof DhtVirginState) {
+		if ((!$this instanceof DhtVirginState) && (!$this instanceof DhtInitState)) {
 			// Throw the exception
 			throw new InvalidStateException($this, self::EXCEPTION_INVALID_STATE);
 		} // END - if