diff --git a/t/001_wal.pl b/t/001_wal.pl
index 5376084..4e9b21c 100644
--- a/t/001_wal.pl
+++ b/t/001_wal.pl
@@ -23,7 +23,8 @@ sub test_index_replay
 	if ($server_version_num >= 100000) {
 		$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() <= replay_location FROM pg_stat_replication WHERE application_name = '$applname';";
+		# TODO figure out why replay location doesn't work
+		$caughtup_query = "SELECT pg_current_xlog_location() <= write_location FROM pg_stat_replication WHERE application_name = '$applname';";
 	}
 
 	$node_primary->poll_query_until('postgres', $caughtup_query)