small modification for a test