Fixed small test warning