Minor bugfix in test code.