diff --git a/internal/2016/DayFour/code_test.go b/internal/2016/DayFour/code_test.go
index 170d850..b75484e 100644
--- a/internal/2016/DayFour/code_test.go
+++ b/internal/2016/DayFour/code_test.go
@@ -1,18 +1,29 @@
 package dayfour
 
-import "testing"
+import (
+	"testing"
+)
 
 var testInput = []string{
 	"aaaaa-bbb-z-y-x-123[abxyz]",
 	"a-b-c-d-e-f-g-h-987[abcde]",
 	"not-a-real-room-404[oarel]",
 	"totally-real-room-200[decoy]",
+	"ijmockjgz-storage-5[gjoac]",
 }
 
 func TestPartOne(t *testing.T) {
-	expected := 1514
+	expected := 1519
 	got := PartOne(testInput)
 	if got != expected {
 		t.Errorf("PartOne() = %d, want %d", got, expected)
 	}
 }
+
+func TestPartTwo(t *testing.T) {
+	expected := 5
+	got := PartTwo(testInput)
+	if got != expected {
+		t.Errorf("PartTwo() = %d, want %d", got, expected)
+	}
+}