diff --git a/dagger/input_test.go b/dagger/input_test.go
index c639576e..a8482946 100644
--- a/dagger/input_test.go
+++ b/dagger/input_test.go
@@ -7,13 +7,16 @@ import (
 )
 
 func TestInputDir(t *testing.T) {
-	st := &RouteState{}
+	st := &RouteState{
+		LayoutSource: DirInput("/tmp/source", []string{}),
+	}
 	require.NoError(t, st.AddInput("www.source", DirInput(".", []string{})))
 
 	route, err := NewRoute(st)
 	require.NoError(t, err)
 
 	localdirs := route.LocalDirs()
-	require.Len(t, localdirs, 1)
-	require.Equal(t, ".", localdirs["."])
+	require.Len(t, localdirs, 2)
+	require.Contains(t, localdirs, ".")
+	require.Contains(t, localdirs, "/tmp/source")
 }