diff --git a/t/001_wal.pl b/t/001_wal.pl
index 35bb5e1..5376084 100644
--- a/t/001_wal.pl
+++ b/t/001_wal.pl
@@ -21,9 +21,9 @@ sub test_index_replay
 	my $caughtup_query;
 	my $server_version_num = $node_primary->safe_psql("postgres", "SHOW server_version_num");
 	if ($server_version_num >= 100000) {
-		$caughtup_query = "SELECT pg_current_wal_lsn() <= write_lsn FROM pg_stat_replication WHERE application_name = '$applname';";
+		$caughtup_query = "SELECT pg_current_wal_lsn() <= replay_lsn FROM pg_stat_replication WHERE application_name = '$applname';";
 	} else {
-		$caughtup_query = "SELECT pg_current_xlog_location() <= write_location FROM pg_stat_replication WHERE application_name = '$applname';";
+		$caughtup_query = "SELECT pg_current_xlog_location() <= replay_location FROM pg_stat_replication WHERE application_name = '$applname';";
 	}
 
 	$node_primary->poll_query_until('postgres', $caughtup_query)