Fix issues in BWT test code (update to align with new interface)