diff --git a/src/pocketmine/entity/Entity.php b/src/pocketmine/entity/Entity.php
index d05bba658..0b2fd4c8b 100644
--- a/src/pocketmine/entity/Entity.php
+++ b/src/pocketmine/entity/Entity.php
@@ -436,7 +436,7 @@ abstract class Entity extends Position implements Metadatable{
 				}
 			}else{
 				if(($this->fireTicks % 20) === 0){
-					$this->attack(1, "onFire");
+					$this->attack(1, EntityDamageEvent::CAUSE_FIRE_TICK);
 				}
 				--$this->fireTicks;
 			}
@@ -444,7 +444,7 @@ abstract class Entity extends Position implements Metadatable{
 		}
 
 		if($this->handleLavaMovement()){
-			$this->attack(4, "lava");
+			$this->attack(4, EntityDamageEvent::CAUSE_LAVA);
 			$this->setOnFire(15);
 			$hasUpdate = true;
 			$this->fallDistance *= 0.5;
@@ -1050,4 +1050,4 @@ abstract class Entity extends Position implements Metadatable{
 		$this->server->getEntityMetadata()->removeMetadata($this, $metadataKey, $plugin);
 	}
 
-}
\ No newline at end of file
+}