diff --git a/integration/buildfile_test.go b/integration/buildfile_test.go
index e5084d4355..f4ed61aaff 100644
--- a/integration/buildfile_test.go
+++ b/integration/buildfile_test.go
@@ -6,6 +6,7 @@ import (
 	"github.com/dotcloud/docker/archive"
 	"github.com/dotcloud/docker/engine"
 	"github.com/dotcloud/docker/image"
+	"github.com/dotcloud/docker/nat"
 	"github.com/dotcloud/docker/utils"
 	"io/ioutil"
 	"net"
@@ -492,7 +493,7 @@ func TestBuildExpose(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	if img.Config.PortSpecs[0] != "4243" {
+	if _, exists := img.Config.ExposedPorts[nat.NewPort("tcp", "4243")]; !exists {
 		t.Fail()
 	}
 }
@@ -594,6 +595,17 @@ func TestBuildImageWithCache(t *testing.T) {
 	checkCacheBehavior(t, template, true)
 }
 
+func TestBuildExposeWithCache(t *testing.T) {
+	template := testContextTemplate{`
+        from {IMAGE}
+        maintainer dockerio
+	expose 80
+	run echo hello
+        `,
+		nil, nil}
+	checkCacheBehavior(t, template, true)
+}
+
 func TestBuildImageWithoutCache(t *testing.T) {
 	template := testContextTemplate{`
         from {IMAGE}
@@ -877,7 +889,7 @@ func TestBuildInheritance(t *testing.T) {
 	}
 
 	// from parent
-	if img.Config.PortSpecs[0] != "4243" {
+	if _, exists := img.Config.ExposedPorts[nat.NewPort("tcp", "4243")]; !exists {
 		t.Fail()
 	}
 }