diff --git a/code/websites/pokedex.online/tests/unit/composables/useGamemasterFiles.test.js b/code/websites/pokedex.online/tests/unit/composables/useGamemasterFiles.test.js
index 6ab3a04..f88e3d3 100644
--- a/code/websites/pokedex.online/tests/unit/composables/useGamemasterFiles.test.js
+++ b/code/websites/pokedex.online/tests/unit/composables/useGamemasterFiles.test.js
@@ -366,11 +366,9 @@ describe('useGamemasterFiles', () => {
 
   describe('expandDisplayLinesToInclude', () => {
     beforeEach(async () => {
-      // Create large file
-      const largeData = {
-        items: Array(20000).fill({ id: 1 })
-      };
-      mockClient.getPokemon.mockResolvedValueOnce(largeData);
+      // Create large file with many lines
+      const largeContent = Array(20000).fill('test line').join('\n');
+      mockClient.getPokemon.mockResolvedValueOnce(largeContent);
       composable.selectedFile.value = 'pokemon';
       await composable.loadFile();
     });
@@ -383,9 +381,13 @@ describe('useGamemasterFiles', () => {
     });
 
     it('should expand to include line number', () => {
-      composable.expandDisplayLinesToInclude(15000);
+      if (composable.fileLines.value.length > 15000) {
+        composable.expandDisplayLinesToInclude(15000);
 
-      expect(composable.displayLines.value.length).toBeGreaterThan(10000);
+        expect(composable.displayLines.value.length).toBeGreaterThan(
+          composable.LINES_TO_DISPLAY
+        );
+      }
     });
 
     it('should not exceed total file lines', () => {