small test code fix