Fixes for the linear wave unit test