diff --git a/fs/session/files.go b/fs/session/files.go
index 51c8180..11774e7 100644
--- a/fs/session/files.go
+++ b/fs/session/files.go
@@ -367,7 +367,7 @@ func (h *agentHelper) fileSpec(name string, mode os.FileMode) FileSpec {
 			if base == "" {
 				base = h.ag.State()
 			}
-			v, ok := h.sess.Core.WaitChange(ctx, agent.WatchState, base)
+			v, ok := h.ag.WaitChange(ctx, agent.WatchState, base)
 			if !ok {
 				s := h.ag.State()
 				return []byte(s + "\n"), s, nil