minor cleanup on test