diff --git a/code/EditableSpamProtectionField.php b/code/EditableSpamProtectionField.php
index f663b64..6c77d98 100644
--- a/code/EditableSpamProtectionField.php
+++ b/code/EditableSpamProtectionField.php
@@ -77,8 +77,11 @@ class EditableSpamProtectionField extends EditableFormField {
 		return "";
 	}
 	
-	function EditSegment() {
-		return $this->renderWith( $this->class );
+	/**
+	 * @return string
+	 */
+	public function Icon() {
+		return 'spamprotection/images/' . strtolower($this->class) . '.png';
 	}
-}
-?>
\ No newline at end of file
+	
+}
\ No newline at end of file
diff --git a/templates/EditableSpamProtectionField.ss b/templates/EditableSpamProtectionField.ss
deleted file mode 100644
index 8ab357d..0000000
--- a/templates/EditableSpamProtectionField.ss
+++ /dev/null
@@ -1,20 +0,0 @@
-<div class="EditableSpamProtectionField EditableFormField" id="$Name.Attr">
-	<div class="FieldInfo">
-		<img class="handle" src="sapphire/images/drag.gif" alt="<% _t('DRAG', 'Drag to rearrange order of fields') %>" />
-		<img class="icon" src="spamprotection/images/editablespamprotectionfield.png" alt="<% _t('SPAMPROTECTIONFIELD', 'Spam Protection Field') %>" />
-		$TitleField
-		<a class="toggler" href="#" title="<% _t('MORE', 'More options') %>"><img src="cms/images/edit.gif" alt="<% _t('MORE', 'More options') %>" /></a>
-		<a class="delete" href="#" title="<% _t('DELETE', 'Delete this field') %>"><img src="cms/images/delete.gif" alt="<% _t('DELETE', 'Delete this field') %>" /></a>
-	</div>
-	<div class="ExtraOptions" id="$Name.Attr-extraOptions">
-		<div class="FieldDefault" id="$Name.Attr-fieldDefault">
-			$DefaultField
-		</div>
-		<% control ExtraOptions %>
-		$FieldHolder
-		<% end_control %>
-	</div>
-    <input type="hidden" name="$Name.Attr[CustomParameter]" value="$CustomParameter" />
-    <input type="hidden" name="$Name.Attr[Type]" value="EditableSpamProtectionField" />
-	<input type="hidden" name="$Name.Attr[Sort]" value="-1" />
-</div>
\ No newline at end of file