Fixed a small test nit