diff --git a/signalgen.py b/signalgen.py
index e1c2d4f..5104193 100755
--- a/signalgen.py
+++ b/signalgen.py
@@ -88,7 +88,7 @@ def addBurst(values):
 	filterbankObj = Your(values.file)
 	values.file = values.file.strip(".fil") #removing extension for text manipulation later
 	basename = path.basename(values.file)
-	samples = filterbankObj.your_header.nsamp
+	samples = filterbankObj.your_header.nspectra
 	
 	#the full filterbanks use 64GB in RAM when injecting burst, so we write out a truncated version and load that instead.
 	filWriter = Writer(filterbankObj, outdir="./", outname = f"{basename}_trunc", nstart = 0, nsamp = samples)
@@ -208,7 +208,7 @@ if __name__ == "__main__":
 		logging.info(f"Running with file {values.file}")
 		if values.plot:
 			filterbankObj = Your(values.file)
-			spectra = filterbankObj.get_data(0, filterbankObj.your_header.nsamp)
+			spectra = filterbankObj.get_data(0, filterbankObj.your_header.nspectra)
 			show_dynamic(spectra, f"{values.file} Dynamic Spectra", save=True)
 		else:
 			addBurst(values)