Fixed a minor issue in a test