Fixed a small test