Fix small test mistake